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

move multi_map to grouper rewrite pass (#10409)

Browse Source 
* move multi_map to grouper rewrite pass

* delete that
This commit is contained in:
qazal
2025-05-19 10:44:06 +03:00
committed by GitHub
parent b06291077c
commit cc8dda1d75
5 changed files with 14 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_multitensor.py
View File

@@ -628,6 +628,9 @@ class TestMultiTensor(unittest.TestCase):
# assert for cannot maintain axis
with self.assertRaises(AssertionError): t0.reshape(4, 3, 2, 7, 15)
# it doesn't work like this anymore
# NOTE: this never failed in assign_multi, it failed tensor spec because MULTI was never pushed in the graph
@unittest.expectedFailure
def test_mlb_assign_change_axis(self):
t_none = Tensor.zeros((16, 16)).shard(devices_2).contiguous().realize()
t_zero = Tensor.ones((16, 16)).shard(devices_2, axis=0)