mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-10-21 06:40:23 +08:00
Lei WangandZhiwen Mo
f8dc61c02c
* test: trim CI runtime for slow Python suites * remove useless test * Update TVM submodule to latest commit and remove manual device assert test from CI. Replace with a call to tilelang.testing.main(). * lint fix * fix * revert test for eager jit --------- Co-authored-by: Zhiwen Mo <zm125@ic.ac.uk>
21 lines
443 B
Python
21 lines
443 B
Python
# type: ignore
|
|
import tilelang
|
|
import tilelang.testing
|
|
import tilelang.language as T
|
|
|
|
|
|
def test_device_assert_no_trigger():
|
|
@T.prim_func
|
|
def program():
|
|
with T.Kernel(threads=128):
|
|
tid = T.get_thread_binding()
|
|
T.device_assert(tid == tid)
|
|
|
|
jit_kernel = tilelang.compile(program)
|
|
profiler = jit_kernel.get_profiler()
|
|
profiler.run_once()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
tilelang.testing.main()
|