mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-10-24 07:04:00 +08:00
Zhengju TangandLei Wang
46eb4589bf
* [Refactor] Enhance TMA barrier validation and support for additional architectures (#463) * Updated the TMA barrier validation in `inject_tma_barrier.cc` to check for non-empty `barrier_id_to_range_` before raising an error for missing `create_list_of_mbarrier`. * Refactored architecture checks in `phase.py` to utilize a new constant `SUPPORTED_TMA_ARCHS`, allowing for easier updates and improved readability in the target architecture validation logic. * [CI] Add BlocksparseGemm, Dynamic, and Cast examples to CI. * Lint --------- Co-authored-by: Lei Wang <34334180+LeiWang1999@users.noreply.github.com>
11 lines
188 B
Python
11 lines
188 B
Python
import tilelang.testing
|
|
import example_blocksparse_gemm
|
|
|
|
|
|
def test_example_blocksparse_gemm():
|
|
example_blocksparse_gemm.main()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
tilelang.testing.main()
|