mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-10-20 06:30:07 +08:00
Chaofan LinandLeiWang1999
1a1df62e94
* [Cleanup] Remove stale deprecation deadlines * [Language] Remove deprecated T.symbolic alias --------- Co-authored-by: LeiWang1999 <leiwang1999@outlook.com>
24 lines
820 B
Python
24 lines
820 B
Python
import tilelang.testing
|
|
from tilelang import language as T
|
|
|
|
|
|
def test_issue_1237_dynamic_copy_extent_builds():
|
|
# Repro from debug/1113_issues/copy_dyn.py, adapted as a unit test.
|
|
# The goal is to ensure T.copy correctly handles dynamic extents
|
|
# (e.g., src slice length vs. static dst buffer size) during prim_func building.
|
|
|
|
length = T.dynamic("len", dtype=T.int32)
|
|
|
|
@T.prim_func
|
|
def sample_kernel(global_tensor: T.Tensor[(length,), T.int32]): # noqa: F821
|
|
with T.Kernel(1, threads=32):
|
|
buffer_shared = T.alloc_shared((1024,), dtype=T.int32)
|
|
T.copy(global_tensor[0:length], buffer_shared)
|
|
|
|
# Building the prim_func is sufficient to exercise the bug path; no need to JIT/execute.
|
|
_ = sample_kernel
|
|
|
|
|
|
if __name__ == "__main__":
|
|
tilelang.testing.main()
|