junrushao commented on code in PR #15142:
URL: https://github.com/apache/tvm/pull/15142#discussion_r1240645514
##########
tests/python/unittest/test_tir_schedule_compute_inline.py:
##########
@@ -622,8 +621,8 @@ def elementwise_overcomputed_producer_reverse_inlined(
for i, j in T.grid(128, 128):
with T.block("B"):
vi, vj = T.axis.remap("SS", [i, j])
- T.where(i < 127 and j < 127)
- C[vi, vj] = A[vi, vj] * 2.0 + 1.0
+ if vi < 127 and vj < 127:
Review Comment:
To be very clear, T.where limits the iteration domain, meaning it’s going to
change the affinity of iteration space, which is not desirable. This PR fixed
this behavior and I count it as a bug fix rather than a breaking change.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
