test_vs_sparse_attn.py 238 Bytes
Newer Older
root's avatar
init  
root committed
1
2
3
4
5
6
7
8
9
10
11
12
import tilelang.testing

import example_vertical_slash_sparse_attn


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


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