slyubomirsky commented on code in PR #16584:
URL: https://github.com/apache/tvm/pull/16584#discussion_r1500974481
##########
tests/python/tir-base/test_tir_specialize.py:
##########
@@ -324,5 +326,41 @@ def expected(A_data: T.handle("float32")):
tvm.ir.assert_structural_equal(expected, after)
+def test_specialization_removes_struct_info():
+ """Reset struct info in specialization
+
+ While a PrimFunc usually doesn't have a `relax.StructInfo`, the
+ field can be populated in some edge cases. If that PrimFunc is
Review Comment:
Always specifying it sounds like a better approach, as long as it would not
interfere with anything in TIR.
--
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: commits-unsubscr...@tvm.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
