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

print tflops in asm gemm test (#13859)

Browse Source 
* print tflops in asm gemm test

* change order
This commit is contained in:
qazal
2025-12-29 02:26:40 +09:00
committed by GitHub
parent a03cd43e78
commit 066d96c397
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
extra/gemm/asm/test.py
View File

@@ -47,7 +47,10 @@ ast = eis[0].ast
prg = get_asm_gemm(ast, fp)
eis.append(ExecItem(ast, [C_asm.uop.buffer, from_torch(B).uop.buffer, from_torch(A).uop.buffer], prg=CompiledRunner(prg)))
for ei in eis: ei.run(wait=True)
for ei in eis:
et = ei.run(wait=True)
print(f"{(N*N*N*2 / et)*1e-12:.2f} REAL TFLOPS")
# ** correctness