This is an automated email from the ASF dual-hosted git repository. ruihangl pushed a commit to branch unity-staging in repository https://gitbox.apache.org/repos/asf/tvm.git
commit da236cb7e90a305915f1b97d08dc33ee8e37645d Author: Ruihang Lai <[email protected]> AuthorDate: Sat Jan 6 17:24:26 2024 -0500 Fix after merging 'main' into 'unity' --- python/tvm/ir/module.py | 1 - tests/python/relax/test_backend_dispatch_sort_scan.py | 6 ++---- tests/python/relax/test_transform_legalize_ops_grad.py | 4 ---- tests/python/relax/test_transform_legalize_ops_manipulate.py | 12 ------------ .../relax/test_transform_legalize_ops_search_statistical.py | 4 ---- 5 files changed, 2 insertions(+), 25 deletions(-) diff --git a/python/tvm/ir/module.py b/python/tvm/ir/module.py index cd13e90701..a3e097947c 100644 --- a/python/tvm/ir/module.py +++ b/python/tvm/ir/module.py @@ -17,7 +17,6 @@ """IRModule that holds the functions and type definitions.""" from __future__ import annotations -import ast from typing import Dict, Union import tvm._ffi diff --git a/tests/python/relax/test_backend_dispatch_sort_scan.py b/tests/python/relax/test_backend_dispatch_sort_scan.py index 79234c2af8..c21dd4882f 100644 --- a/tests/python/relax/test_backend_dispatch_sort_scan.py +++ b/tests/python/relax/test_backend_dispatch_sort_scan.py @@ -50,8 +50,6 @@ def test_dispatch_cumsum(): T.func_attr({"tir.noalias": T.bool(True)}) A = T.match_buffer(var_A, (T.int64(2), T.int64(3)), offset_factor=1) with T.block("cumsum_generic"): - T.reads(A[T.int64(0) : T.int64(2), T.int64(0) : T.int64(3)]) - T.writes(out_buf[T.int64(0) : T.int64(2), T.int64(0) : T.int64(3)]) for fused in T.parallel(T.int64(2)): out_buf[ fused * T.int64(3) // T.int64(3), fused * T.int64(3) % T.int64(3) @@ -150,8 +148,8 @@ def test_dispatch_sort(): data_buf = T.match_buffer(var_A, (m, T.int64(3)), align=8) out_buf = T.match_buffer(var_sort_cpu, (m, T.int64(3)), align=8) with T.block("sort_cpu"): - T.reads(data_buf[T.int64(0) : m, T.int64(0) : T.int64(3)]) - T.writes(out_buf[T.int64(0) : m, T.int64(0) : T.int64(3)]) + T.reads() + T.writes() T.call_packed( "tvm.contrib.sort.sort", T.tvm_stack_make_array( diff --git a/tests/python/relax/test_transform_legalize_ops_grad.py b/tests/python/relax/test_transform_legalize_ops_grad.py index f38e1f971e..19d1a106f8 100644 --- a/tests/python/relax/test_transform_legalize_ops_grad.py +++ b/tests/python/relax/test_transform_legalize_ops_grad.py @@ -313,8 +313,6 @@ def test_take_backward(): rxplaceholder_1 = T.match_buffer(var_rxplaceholder_1, (T.int64(3), T.int64(4), T.int64(5)), offset_factor=1) rxplaceholder_2 = T.match_buffer(var_rxplaceholder_2, (T.int64(2),), "int32", offset_factor=1) with T.block("take_backward"): - T.reads(rxplaceholder[T.int64(0):T.int64(3), T.int64(0):T.int64(2), T.int64(0):T.int64(5)], rxplaceholder_1[T.int64(0):T.int64(3), T.int64(0):T.int64(4), T.int64(0):T.int64(5)], rxplaceholder_2[T.int64(0):T.int64(2)]) - T.writes(out_buf[T.int64(0):T.int64(3), T.int64(0):T.int64(4), T.int64(0):T.int64(5)]) for i in range(T.int64(60)): out_buf[i // T.int64(5) // T.int64(4), i // T.int64(5) % T.int64(4), i % T.int64(5)] = T.float32(0) for parallel, serial in T.grid(T.int64(15), T.int64(2)): @@ -354,8 +352,6 @@ def test_take_backward_symbolic(): rxplaceholder_2 = T.match_buffer(var_rxplaceholder_2, (i,), "int32", offset_factor=1) out_buf = T.match_buffer(var_take_backward, (m, n)) with T.block("take_backward"): - T.reads(rxplaceholder[T.int64(0):m, T.int64(0):i], rxplaceholder_1[T.int64(0):m, T.int64(0):n], rxplaceholder_2[T.int64(0):i]) - T.writes(out_buf[T.int64(0):m, T.int64(0):n]) for i_1 in range(m * n): out_buf[i_1 // n, i_1 % n] = T.float32(0) for parallel, serial in T.grid(m, i): diff --git a/tests/python/relax/test_transform_legalize_ops_manipulate.py b/tests/python/relax/test_transform_legalize_ops_manipulate.py index 6c9ca9d980..9b7a8f23c9 100644 --- a/tests/python/relax/test_transform_legalize_ops_manipulate.py +++ b/tests/python/relax/test_transform_legalize_ops_manipulate.py @@ -1352,12 +1352,6 @@ def test_scatter_elements(): var_rxplaceholder_2, (T.int64(2), T.int64(2)), offset_factor=1 ) with T.block("scatter_elements_generic"): - T.reads( - rxplaceholder[T.int64(0) : T.int64(4), T.int64(0) : T.int64(4)], - rxplaceholder_1[T.int64(0) : T.int64(2), T.int64(0) : T.int64(2)], - rxplaceholder_2[T.int64(0) : T.int64(2), T.int64(0) : T.int64(2)], - ) - T.writes(out_buf[T.int64(0) : T.int64(4), T.int64(0) : T.int64(4)]) for i in T.parallel(T.int64(16)): out_buf[i // T.int64(4), i % T.int64(4)] = rxplaceholder[ i // T.int64(4), i % T.int64(4) @@ -1453,12 +1447,6 @@ def test_scatter_elements_symbolic(): rxplaceholder_2 = T.match_buffer(var_rxplaceholder_2, (m, n), offset_factor=1) out_buf = T.match_buffer(var_scatter_elements_generic, (a, b)) with T.block("scatter_elements_generic"): - T.reads( - rxplaceholder[T.int64(0) : a, T.int64(0) : b], - rxplaceholder_1[T.int64(0) : m, T.int64(0) : n], - rxplaceholder_2[T.int64(0) : m, T.int64(0) : n], - ) - T.writes(out_buf[T.int64(0) : a, T.int64(0) : b]) for i in T.parallel(a * b): out_buf[i // b, i % b] = rxplaceholder[i // b, i % b] for fused in T.parallel(m): diff --git a/tests/python/relax/test_transform_legalize_ops_search_statistical.py b/tests/python/relax/test_transform_legalize_ops_search_statistical.py index 979f3f113b..c6c53ff0b9 100644 --- a/tests/python/relax/test_transform_legalize_ops_search_statistical.py +++ b/tests/python/relax/test_transform_legalize_ops_search_statistical.py @@ -1082,8 +1082,6 @@ def test_cumsum(): T.func_attr({"tir.noalias": True}) rxplaceholder = T.match_buffer(var_rxplaceholder, (T.int64(3), T.int64(2), T.int64(3)), offset_factor=1) with T.block("cumsum_generic"): - T.reads(rxplaceholder[T.int64(0):T.int64(3), T.int64(0):T.int64(2), T.int64(0):T.int64(3)]) - T.writes(out_buf[T.int64(0):T.int64(3), T.int64(0):T.int64(2), T.int64(0):T.int64(3)]) for fused in T.parallel(T.int64(9)): out_buf[(fused // T.int64(3) * T.int64(2) * T.int64(3) + fused % T.int64(3)) // T.int64(3) // T.int64(2), (fused // T.int64(3) * T.int64(2) * T.int64(3) + fused % T.int64(3)) // T.int64(3) % T.int64(2), (fused // T.int64(3) * T.int64(2) * T.int64(3) + fused % T.int64(3)) % T.int64(3)] = T.Cast("int32", rxplaceholder[(fused // T.int64(3) * T.int64(2) * T.int64(3) + fused % T.int64(3)) // T.int64(3) // T.int64(2), (fused // T.int64(3) * T.int64(2) * T.int64(3) + fused % [...] for _k in range(T.int64(1)): @@ -1118,8 +1116,6 @@ def test_cumsum_symbolic(): rxplaceholder = T.match_buffer(var_rxplaceholder, (a, b, c), offset_factor=1) out_buf = T.match_buffer(var_cumsum_generic, (a, b, c), "int32") with T.block("cumsum_generic"): - T.reads(rxplaceholder[T.int64(0):a, T.int64(0):b, T.int64(0):c]) - T.writes(out_buf[T.int64(0):a, T.int64(0):b, T.int64(0):c]) for fused in T.parallel(a * c): out_buf[(fused // c * b * c + fused % c) // c // b, (fused // c * b * c + fused % c) // c % b, (fused // c * b * c + fused % c) % c] = T.Cast("int32", rxplaceholder[(fused // c * b * c + fused % c) // c // b, (fused // c * b * c + fused % c) // c % b, (fused // c * b * c + fused % c) % c]) for _k in range(b - T.int64(1)):
