test_vs_sparse_attn.py 238 Bytes
Newer Older
1
2
3
4
5
6
7
import tilelang.testing

import example_vertical_slash_sparse_attn


@tilelang.testing.requires_cuda
def test_vs_sparse_attn():
8
    example_vertical_slash_sparse_attn.main(argv=[])
9
10
11


if __name__ == "__main__":
12
    tilelang.testing.main()