Unverified Commit 71b73e18 authored by Kuris's avatar Kuris Committed by GitHub
Browse files

[Refactor] Disable strided buffer load inside tvm (#1301) (#1332)

parent b0206854
Subproject commit 3354ada79dd428e383102020814fa9c37638e752 Subproject commit e3af400013551755a8df668ba77b530735931ade
...@@ -427,7 +427,7 @@ def test_var_macro(): ...@@ -427,7 +427,7 @@ def test_var_macro():
pass pass
def frame_inside_macro(): def test_frame_inside_macro():
@tilelang.jit @tilelang.jit
def get_sample_kernel(): def get_sample_kernel():
...@@ -453,5 +453,18 @@ def frame_inside_macro(): ...@@ -453,5 +453,18 @@ def frame_inside_macro():
kernel = get_sample_kernel() # noqa: F841 kernel = get_sample_kernel() # noqa: F841
def test_buffer_slice_step():
try:
@T.prim_func
def prim_buffer_slice_step(A: T.Buffer((10,), T.int32), B: T.Buffer((5,), T.int32)):
with T.Kernel(1):
B[0:5:2] = A[0:10:2]
raise AssertionError("Expect to report an error, buffer slice with step is not supported")
except RuntimeError:
pass
if __name__ == '__main__': if __name__ == '__main__':
tilelang.testing.main() tilelang.testing.main()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment