Skip to content

Conversation

ndrego
Copy link
Contributor

@ndrego ndrego commented Oct 3, 2023

Description

Changes symbolic shape inferencing for QuadricCustomOp to handle multiple outputs

Motivation and Context

Symbolic shape inferencing for QuadricCustomOp assumed a single output

@ndrego ndrego force-pushed the nigel-quadric-custom-shape-inference branch from 90b3815 to 138bb6b Compare October 3, 2023 17:54
elif node.op_type == "QuadricCustomOp":
# Should have a subgraph, but allow for cases where it's not there
subgraph = get_attribute(node, "sub_graph")
subgraphs = [subgraph] if subgraph else []
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

What happens when there is no subgraph found in the QuadricCustomOp? Do we need to assert or let it be?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

For now we can let it be. There is some effort required to change over some old ONNXs found in TVM tests, which is why I'm letting it go for now. Since we're not currently using the subgraph for symbolic shape inference, there's no impact here.

@veerbhan veerbhan self-requested a review October 3, 2023 22:54
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants