This is an automated email from the ASF dual-hosted git repository.
kparzysz pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new b84ed27260 [TIR][BugFix] Fix a wrong use of T.exp in
test_compute_inline_opaque_access_with_tvm_access_ptr. (#12117)
b84ed27260 is described below
commit b84ed27260d492e954963448e6cc8aee329b4cc6
Author: albert qing <2628...@qq.com>
AuthorDate: Sun Jul 17 06:11:02 2022 +0800
[TIR][BugFix] Fix a wrong use of T.exp in
test_compute_inline_opaque_access_with_tvm_access_ptr. (#12117)
Co-authored-by: sqing <qing.s...@intellif.com>
---
tests/python/unittest/test_tir_schedule_compute_inline.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/tests/python/unittest/test_tir_schedule_compute_inline.py
b/tests/python/unittest/test_tir_schedule_compute_inline.py
index 617e13db27..ec19402969 100644
--- a/tests/python/unittest/test_tir_schedule_compute_inline.py
+++ b/tests/python/unittest/test_tir_schedule_compute_inline.py
@@ -556,11 +556,11 @@ def exp_exp_opaque_access_with_tvm_access_ptr(
for i0 in T.serial(16):
with T.block("compute_1"):
i0_2 = T.axis.spatial(16, i0)
- T.reads(compute_1[i0_2], lookup_table[0:1024])
+ T.reads(lookup_table[0:1024], compute_1[i0_2])
T.writes(compute[i0_2])
+ T.evaluate(lookup_table.access_ptr("r"))
compute[i0_2] = T.exp(
compute_1[i0_2],
- lookup_table.access_ptr("r"),
dtype="float16",
)
@@ -576,11 +576,11 @@ def exp_exp_opaque_access_with_tvm_access_ptr_inlined(
i0_1 = T.axis.spatial(16, i0)
# Do not put the opaque access to new write region when opaque
access
# wrapped with a tvm_access_ptr and the access mask set to "read
only"
- T.reads(x[i0_1], lookup_table[0:1024])
+ T.reads(lookup_table[0:1024], x[i0_1])
T.writes(compute[i0_1])
+ T.evaluate(lookup_table.access_ptr("r"))
compute[i0_1] = T.exp(
T.exp(x[i0_1], dtype="float16"),
- lookup_table.access_ptr("r"),
dtype="float16",
)
