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

isolate slow llama gemm (#14525)

Browse Source
This commit is contained in:
qazal
2026-02-03 22:20:10 -05: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

1
extra/gemm/asm/cdna/test_asm_gemm.py
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()