Skip to content

Fix onnx round import with float64 inputs.#11685

Merged
AndrewZhaoLuo merged 2 commits intoapache:mainfrom
lazycal:fix-onnx-round-float64
Jun 13, 2022
Merged

Fix onnx round import with float64 inputs.#11685
AndrewZhaoLuo merged 2 commits intoapache:mainfrom
lazycal:fix-onnx-round-float64

Conversation

@lazycal
Copy link
Contributor

@lazycal lazycal commented Jun 12, 2022

This simple PR fixes the unhandled float64 case introduced by #11446. Without the change, TVM won't be able to import any ONNX Round operator with float64 inputs, and will complain with error data types float64 and float32 do not match in BroadcastRel.
cc: @AndrewZhaoLuo @jwfromm @CircleSpin

Copy link
Contributor

@AndrewZhaoLuo AndrewZhaoLuo left a comment

Choose a reason for hiding this comment

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

Thanks @lazycal

@AndrewZhaoLuo AndrewZhaoLuo merged commit 85a190a into apache:main Jun 13, 2022
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.

2 participants