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

removed unnecessary 'isinstance(data, UOp)' check (#10605)

Browse Source
This commit is contained in:
ihar
2025-06-02 19:58:14 -05:00
committed by GitHub
parent 07de095b27
commit ba02a6331e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/tensor.py
View File

@@ -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