Skip to content

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