geohot/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Actions Activity

isolate slow llama gemm (#14525)

Browse Source
This commit is contained in:
qazal
2026-02-04 12:20:10 +09:00
committed by GitHub
parent 2f55005ad9
commit d1bfbe9ce3
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extra/gemm/asm/cdna/test_asm_gemm.py
+1
View File
@@ -48,6 +48,7 @@ class TestGemmLarge(unittest.TestCase):
def test_gemm4(self): verify_asm_gemm(8, 4096, 14336, 4096, gpus=8)
def test_gemm5(self): verify_asm_gemm(8, 4096, 4096, 14336, gpus=8)
def test_gemm6(self): verify_asm_gemm(16, 4096, 4096, 14336, gpus=8)
def test_gemm7(self): verify_asm_gemm(1, 8192, 128256, 4096)
if __name__ == "__main__":
unittest.main()