mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-10-24 07:04:00 +08:00
Kuris
5358f5a272
* [Refactor] Support lazy evaluate in eager jit * add test script * fix lint error * canonicalize signature and error of `T.empty` * Fix comment from genai * [EagerJIT] Add comment for eager jit stage
22 lines
477 B
Python
22 lines
477 B
Python
import tilelang
|
|
import tilelang.testing
|
|
import tilelang.language as T
|
|
|
|
|
|
def test_issue_1690():
|
|
@tilelang.jit()
|
|
def test(A):
|
|
N = T.const("N")
|
|
A: T.Tensor[[N], T.float32]
|
|
with T.Kernel():
|
|
tmp = T.alloc_fragment((N,), T.float32)
|
|
tmp_max = T.alloc_fragment(1, T.float32)
|
|
T.copy(A, tmp)
|
|
T.reduce_max(tmp, tmp_max, dim=0)
|
|
|
|
test.compile(N=16)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
tilelang.testing.main()
|