diff --git a/tinygrad/uop/ops.py b/tinygrad/uop/ops.py index d62f9c9db2385..e6f835a8c05bb 100644 --- a/tinygrad/uop/ops.py +++ b/tinygrad/uop/ops.py @@ -448,8 +448,6 @@ def broadcast(self, count:int): if count == 1: return self return UOp(Ops.STACK, self.dtype.vec(count), (self,)*count) def cast(self, dtype:DType): - # TODO: we shouldn't have to check for dtype.count == 1 here, but CAST is misused in AMD LLVM - if dtype.count == 1 and dtype.count != self.dtype.count: dtype = dtype.vec(self.dtype.count) if self.dtype == dtype: return self return UOp(Ops.CAST, dtype, (self,)) def bitcast(self, dtype:DType): return self if self.dtype == dtype else UOp(Ops.BITCAST, dtype, (self,))