github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-07 03:00:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

remove getattr [run_process_replay] (#5370)

Browse Source 
* remove getattr [run_process_replay]

* don't waste lines
This commit is contained in:
George Hotz
2024-07-10 19:42:17 -07:00
committed by GitHub
parent 0215c952c5
commit 909ad72c53
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tinygrad/codegen/uops.py
View File

@@ -1,5 +1,5 @@
from __future__ import annotations
from typing import Optional, Tuple, Any, Set, cast, List
from typing import Optional, Tuple, Any, Set, cast, List, Union
import functools
from enum import Enum, auto
from dataclasses import dataclass
@@ -62,12 +62,10 @@ class UOp:
def ge(self, x): return -self.lt(x)
def max(self, x): return UOp.alu(BinaryOps.MAX, self, x)
def min(self, x): return -UOp.alu(BinaryOps.MAX, -self, -x)
def __getattribute__(self, attr):
if attr == "const": return functools.partial(UOp.const, self.dtype)
return super().__getattribute__(attr)
def const(self:Union[UOp, DType, None], b:ConstType|Variable): return UOp._const(self.dtype if isinstance(self, UOp) else self, b)
@staticmethod
@functools.lru_cache(maxsize=None)
def const(dtype:Optional[DType], b:ConstType|Variable):
def _const(dtype:Optional[DType], b:ConstType|Variable):
if isinstance(b, Variable): return UOp(UOps.DEFINE_VAR, dtype, (), b)
return UOp(UOps.CONST, dtype, arg=dtypes.as_const(b, dtype) if dtype is not None else b)
@staticmethod