mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-10-26 07:22:54 +08:00
senlyu163
f4f87f468d
* Remove JIT decorator from elementwise_add function in examples * fix kernel compilation without autotune * Refactor main function to accept parameters and update tests for autotune option * Refactor autotune test function for morden style
15 lines
284 B
Python
15 lines
284 B
Python
import tilelang.testing
|
|
import example_elementwise_add
|
|
|
|
|
|
def test_example_elementwise_add():
|
|
example_elementwise_add.main()
|
|
|
|
|
|
def test_example_elementwise_add_autotune():
|
|
example_elementwise_add.main(use_autotune=True)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
tilelang.testing.main()
|