github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-29 00:38:10 -05:00
Code Issues Packages Projects Releases Wiki Activity

dont unbind in ops

Browse Source
This commit is contained in:
qazal
2024-10-31 11:33:56 +02:00
parent a0bd385448
commit e8d9da936d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/ops.py
View File

@@ -330,7 +330,7 @@ class UOp(MathTrait, metaclass=UOpMetaClass):
return UOp(UOps.ALU, out_dtype, (self,)+src, arg)
@staticmethod
def const(dtype:DType, b:Tuple[ConstType, ...]|ConstType|Variable):
if isinstance(b, UOp): return b.unbind()[0] if b.op is UOps.BIND else b
if isinstance(b, UOp): return b
if isinstance(b, tuple) and all_same(b): b = b[0] # doesn't have to be a VCONST if they are all the same
return UOp(UOps.VCONST if isinstance(b, tuple) else UOps.CONST, dtype, arg=dtypes.as_const(b, dtype) if dtype is not None else b) # type: ignore
@staticmethod