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

const copy folding spec + multi.py behavior [pr] (#8436)

Browse Source 
* const copy folding spec + multi behavior [pr]

* copy from clang, move multi test
This commit is contained in:
qazal
2024-12-29 17:12:13 +02:00
committed by GitHub
parent a44cd1e6f7
commit 34987a03af
5 changed files with 32 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_const_folding.py
View File

@@ -240,6 +240,8 @@ class TestMultiConstFolding(unittest.TestCase):
_check_ast_count(0, t ** 1)
_check_ast_count(0, 1 ** t)
# failing because multi calls .contiguous() on every single sharded uop
@unittest.expectedFailure
def test_multi_const_folding_tensor(self):
ds = tuple(f"{Device.DEFAULT}:{i}" for i in range(4))
t = Tensor.arange(16).float().realize().to(ds)