From ba02a6331e25e754651b7f62b1a0eb0f5417fcf9 Mon Sep 17 00:00:00 2001 From: ihar Date: Mon, 2 Jun 2025 19:58:14 -0500 Subject: [PATCH] removed unnecessary 'isinstance(data, UOp)' check (#10605) --- tinygrad/tensor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/tensor.py b/tinygrad/tensor.py index b26068797e..7e55214099 100644 --- a/tinygrad/tensor.py +++ b/tinygrad/tensor.py @@ -163,7 +163,7 @@ class Tensor(MathTrait): # data might be on a different device if isinstance(device, str): self.lazydata:UOp = data if data.device == device else data.copy_to_device(device) # if device is a tuple, we should have/construct a MultiLazyBuffer - elif isinstance(data, UOp) and isinstance(data.device, str): self.lazydata = Tensor(data).shard(device).lazydata + elif isinstance(data.device, str): self.lazydata = Tensor(data).shard(device).lazydata else: assert data.device == device, f"MultiLazyBuffer device mismatch, {data.device} != {device}" self.lazydata = data