testing/python/jit/test_tilelang_jit_gemm.py · 38ba083b581d3f9f1424e1bcfd45ca068bce65cd · OpenDAS / tilelang

[Dev] Implement test case for tilelang transformations (#53) · 38ba083b

Lei Wang authored Jan 25, 2025

* implement jit test case

* [Dev] implement auto tune test case for matrix multiplication

* Implement test for legalize memory access and vectorized loop

* lint fix

38ba083b

test_tilelang_jit_gemm.py 5.85 KB

Replace test_tilelang_jit_gemm.py