This is an automated email from the ASF dual-hosted git repository.
junrushao pushed a commit to branch unity
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/unity by this push:
new 82ee2164d1 [Unity] Update test cases of LazyTransformParams as impure
(#15803)
82ee2164d1 is described below
commit 82ee2164d132542df59597d3c6959ee8d96d4c8c
Author: Wuwei Lin <[email protected]>
AuthorDate: Thu Sep 21 14:02:53 2023 -0700
[Unity] Update test cases of LazyTransformParams as impure (#15803)
Mark the rest of test cases as impure. This is a follow up of #15697
---
tests/python/relax/test_transform_lazy_transform_params.py | 6 ++----
1 file changed, 2 insertions(+), 4 deletions(-)
diff --git a/tests/python/relax/test_transform_lazy_transform_params.py
b/tests/python/relax/test_transform_lazy_transform_params.py
index 5eac747d18..9ecebafae6 100644
--- a/tests/python/relax/test_transform_lazy_transform_params.py
+++ b/tests/python/relax/test_transform_lazy_transform_params.py
@@ -239,9 +239,8 @@ def test_param_shape_symbolic():
T.writes(out[o, i, h, w])
out[o, i, h, w] = w1[i, o, h, w]
- @R.function
+ @R.function(pure=False)
def main_transform_params() -> R.Tuple:
- R.func_attr({"relax.force_pure": True})
ic = T.int64()
cls = Expected
gv: R.Object = R.call_packed("get_item", R.prim_value(1),
sinfo_args=(R.Object,))
@@ -302,9 +301,8 @@ def test_output_with_use_site():
T.writes(y[()])
y[()] = x[()]
- @R.function
+ @R.function(pure=False)
def main_transform_params() -> R.Tuple:
- R.func_attr({"relax.force_pure": True})
cls = Expected
x: R.Object = R.call_packed("get_item", R.prim_value(0),
sinfo_args=(R.Object,))
gv: R.Tensor((), dtype="float32") = R.match_cast(x, R.Tensor((),
dtype="float32"))
