diff --git a/3rdparty/tvm b/3rdparty/tvm index 3354ada79..e3af40001 160000 --- a/3rdparty/tvm +++ b/3rdparty/tvm @@ -1 +1 @@ -Subproject commit 3354ada79dd428e383102020814fa9c37638e752 +Subproject commit e3af400013551755a8df668ba77b530735931ade diff --git a/testing/python/language/test_tilelang_language_frontend_v2.py b/testing/python/language/test_tilelang_language_frontend_v2.py index 349f3cafd..299a41270 100644 --- a/testing/python/language/test_tilelang_language_frontend_v2.py +++ b/testing/python/language/test_tilelang_language_frontend_v2.py @@ -427,7 +427,7 @@ def prim_call_macro(): pass -def frame_inside_macro(): +def test_frame_inside_macro(): @tilelang.jit def get_sample_kernel(): @@ -453,5 +453,18 @@ def sample_kernel( 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__': tilelang.testing.main()