test_example_elementwise.py 284 Bytes
Newer Older
1
2
3
4
5
6
7
8
import tilelang.testing
import example_elementwise_add


def test_example_elementwise_add():
    example_elementwise_add.main()


9
10
11
12
def test_example_elementwise_add_autotune():
    example_elementwise_add.main(use_autotune=True)


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