anwang2009 commented on a change in pull request #8985:
URL: https://github.com/apache/tvm/pull/8985#discussion_r706469124
##########
File path: python/tvm/relay/op/_transform.py
##########
@@ -182,6 +182,12 @@ def compute_unique(attrs, inputs, output_type):
_reg.register_strategy("invert_permutation",
strategy.invert_permutation_strategy)
_reg.register_shape_func("invert_permutation", False, elemwise_shape_func)
+
+# einsum
+_reg.register_strategy("einsum", strategy.einsum_strategy)
+_reg.register_shape_func("einsum", False, elemwise_shape_func)
Review comment:
I'll just remove the shape registration line for now, if we ever
encounter a model with einsum shape dynamism I'll take a look at this again
--
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
