jcf94 commented on a change in pull request #7898:
URL: https://github.com/apache/tvm/pull/7898#discussion_r619056528
##########
File path: tests/python/unittest/test_te_schedule.py
##########
@@ -321,6 +321,33 @@ def test_legalize_invalid_attach():
assert isinstance(stmt, tvm.tir.stmt.For)
+def test_compute_at():
+ def add():
+ shape = (16, 16)
+ A = tvm.te.compute(shape, lambda *i: 1.0, name="A")
+ B = tvm.te.compute(shape, lambda *i: 2.0, name="B")
+ C = tvm.te.compute(shape, lambda *i: A(*i) + B(*i), name="C")
+ return A, B, C
+
+ def invalid_compute_at_self():
+ A, B, C = add()
+ s = tvm.te.create_schedule(C.op)
+ s[C].compute_at(s[C], C.op.axis[0])
Review comment:
Is there any way that we can just dismiss this operation?
It's fine to raise error here, but I'm thinking that if we can treat
compute_at an axis to its self as no effect would be better.
--
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.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
