mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-10-24 07:04:00 +08:00
Lei Wang
c9cec62557
* fix * lint fix * Remove print statement in test_tilelang_issue_1678.py Remove print statement for kernel source in test.
27 lines
610 B
Python
27 lines
610 B
Python
# ruff: noqa
|
|
import tilelang
|
|
import tilelang.testing
|
|
import tilelang.language as T
|
|
|
|
|
|
def test_issue_1678():
|
|
@tilelang.jit
|
|
def qwq():
|
|
@T.prim_func
|
|
def qwq_kernel():
|
|
with T.Kernel(4096, 1, threads=1) as (pid_y, pid_x):
|
|
i = T.alloc_var("int32")
|
|
i = 1
|
|
tmp_row = T.alloc_local((4,), "float32")
|
|
amax_local = T.alloc_var("float32")
|
|
j = 0
|
|
amax_local = T.max(amax_local, tmp_row[j])
|
|
|
|
return qwq_kernel
|
|
|
|
kernel = qwq()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
tilelang.testing.main()
|