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

do not use subbuffer for bitcast (#8514)

Browse Source 
* do not use subbuffer for bitcast

* edit that test

* explicit test for ptx

* ptx
This commit is contained in:
qazal
2025-01-06 18:40:46 +02:00
committed by GitHub
parent 280143467b
commit ed618a72e7
3 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/test_schedule.py
View File

@@ -1334,13 +1334,12 @@ class TestSchedule(unittest.TestCase):
p = np.tile(p, 2)
np.testing.assert_allclose(tiny_ret, p)
@unittest.skipIf(Device.DEFAULT not in view_supported_devices, "subbuffer not supported")
def test_bitcast_subbufer(self):
def test_bitcast_fuses(self):
x = cast(UOp, Tensor.empty(1, dtype=dtypes.float32).realize().lazydata)
a = x.alu(Ops.EXP2).cast(dtypes.int32, True)
b = x.cast(dtypes.int32, True)
b = a.alu(Ops.ADD, b)
check_schedule(b, 2) # this should fuse when it makes sense
check_schedule(b, 1) # this should fuse when it makes sense
@unittest.skip("disabling subbuffer manually isn't supported anymore")
def test_bitcast_disable_subbufer(self):