Skip to content

[Relay][Frontend][Onnx] Fix GEMM converter when C is not a parameter. - #7509

Merged
masahi merged 3 commits into
apache:mainfrom
jwfromm:onnx_gemm_fix
Feb 24, 2021
Merged

[Relay][Frontend][Onnx] Fix GEMM converter when C is not a parameter.#7509
masahi merged 3 commits into
apache:mainfrom
jwfromm:onnx_gemm_fix

Conversation

@jwfromm

@jwfromm jwfromm commented Feb 23, 2021

Copy link
Copy Markdown
Contributor

The change made to the Onnx gemm converter in #7489 assumes that the value of C will always be a parameter. When it is a constant or an expression, the special case handling will fail. This very tiny PR makes the handling of C more general and adds test cases that would otherwise fail.

@jwfromm

jwfromm commented Feb 23, 2021

Copy link
Copy Markdown
Contributor Author

@mbrookhart @xutianming @masahi can one of you take a look at this PR?

@mbrookhart mbrookhart left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

LGTM

@masahi
masahi merged commit d425c14 into apache:main Feb 24, 2021
Lokiiiiii pushed a commit to Lokiiiiii/tvm that referenced this pull request Mar 2, 2021
…apache#7509)

* Fix onnx gemm with non parameter C.

* Add gemm tests for C.

* Fix formatting.
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Mar 2, 2021
…apache#7509)

* Fix onnx gemm with non parameter C.

* Add gemm tests for C.

* Fix formatting.
@jwfromm
jwfromm deleted the onnx_gemm_fix branch April 12, 2023 15:55
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.

3 participants