Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[BugFix]: select node type error in NarrowDataType pass #10519

Merged
merged 1 commit into from
Mar 8, 2022

Conversation

ganler
Copy link
Contributor

@ganler ganler commented Mar 7, 2022

Similar issues as:
#10172
#9582

The select node requires that ture_value.dtype() == false_value.dtype(). While sometimes it is not the case when doing "Narrow Datatype" optimization. This usually happens when we have an argmin / argmax node with some other nodes. The workaround is to make true_value and false_value type-compatible before creating a Select node.

An artificial ONNX model triggering this bug is attached: model.zip

cc: @masahi

@masahi masahi merged commit 71d6f46 into apache:main Mar 8, 2022
@ganler ganler deleted the fix-select-narrow-dtype branch March 8, 2022 20:31
ziqiangxu8457 pushed a commit to ziqiangxu8457/tvm that referenced this pull request Mar 9, 2022
pfk-beta pushed a commit to pfk-beta/tvm that referenced this pull request Apr 11, 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