mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-11-01 08:23:24 +08:00
Wenhao XieandLei Wang
3f25bd1bb1
* add convolution example to CI * lint fix * Update test_example_convolution.py * fix bug --------- Co-authored-by: Lei Wang <34334180+LeiWang1999@users.noreply.github.com>
13 lines
208 B
Python
13 lines
208 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()
|