JinjieLiu/triton-tvm-ffi
1
0 Fork 0
mirror of https://github.com/sgjzfzzf/triton-tvm-ffi.git synced 2026-10-29 03:43:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

support attention fwd

Browse Source 
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
This commit is contained in:
JinjieLiu committed 2026-02-09 17:15:51 +08:00
1 parent 213e4fc060
commit e41ec26329
3 files changed
+904 -1
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

examples/mm/mm.py
+1 -1
View File
@@ -303,5 +303,5 @@ if __name__ == "__main__":
matmul(a, b, "")
cp3 = time.perf_counter_ns()
print(
f"PyTorch matmul: {(cp1 - cp0) / round * 1e-6:.3f} ms\nTriton matmul: {(cp2 - cp1) / round * 1e-6:.3f} ms\nTVM FFI matmul: {(cp3 - cp2) / round * 1e-6:.3f} ms"
f"PyTorch: {(cp1 - cp0) / round * 1e-6:.3f} ms\nTriton: {(cp2 - cp1) / round * 1e-6:.3f} ms\nTVM FFI: {(cp3 - cp2) / round * 1e-6:.3f} ms"
)