tinygrad/tinygrad
2
0
Fork 3
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-08-29 12:16:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

broken

Browse Source
This commit is contained in:
geohot
2025-10-01 17:27:17 +08:00
parent 8e6126160f
commit 3dc1b2e98e
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/test_rangeify.py
+3 -3
View File
@@ -84,13 +84,13 @@ class TestRangeify(unittest.TestCase):
C = Tensor.empty(N, N)
(A@B@C).realize()
@unittest.skip("gemm tc")
#@unittest.skip("gemm tc")
def test_double_gemm_tc(self):
with Context(DEBUG=0):
A, B, C = [Tensor.randn(N, N) for _ in range(3)]
Tensor.realize(A, B, C)
args = (Opt(OptOps.TC, 0, (0,0,1,1))), Opt(OptOps.TC, 0, (0,0,1,0))
#args = (Opt(OptOps.TC, 0, (0,0,1,0)),)
#args = (Opt(OptOps.TC, 0, (0,0,1,1))), Opt(OptOps.TC, 0, (0,0,1,0))
args = (Opt(OptOps.TC, 0, (0,0,1,0)),)
#args = (Opt(OptOps.TC, 0, (0,0,1,1)),)
tst = (A@B@C).contiguous(arg=args).realize()
assert tst.uop.base.op is Ops.BUFFER, "buffer"