mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-10-24 07:04:00 +08:00
Wenhao XieandLei Wang
abe170a639
* add convolution example to CI * lint fix * Update test_example_convolution.py --------- Co-authored-by: Lei Wang <34334180+LeiWang1999@users.noreply.github.com>
13 lines
206 B
Python
13 lines
206 B
Python
import tilelang.testing
|
|
|
|
import example_convolution
|
|
|
|
|
|
@tilelang.testing.requires_cuda
|
|
def test_example_convolution():
|
|
example_convolution.main()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
tilelang.testing.main()
|