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

[BUG] Tensor intrinsic dtype mismatch compiling quantized depthwise convolution on arm_cpu target #16533

Closed
lhutton1 opened this issue Feb 7, 2024 · 0 comments
Labels
needs-triage PRs or issues that need to be investigated by maintainers to find the right assignees to address it topi python/tvm/topi type: bug

Comments

@lhutton1
Copy link
Contributor

lhutton1 commented Feb 7, 2024

Expected behaviour:

When an arm_cpu target is used, the model should compile successfully without an error.

Actual behaviour:

When an arm_cpu target is used, the model fails to compile during tensorization

E           at /workspaces/tvm/src/tvm/src/te/operation/tensorize.cc:334
E     File "/workspaces/tvm/src/tvm/src/te/operation/tensorize.cc", line 334
E   TVMError: Failed to match the data type with TensorIntrin tensor_intrin's declaration  provided=int64, intrin=int32

Environment:

Tested with TVM at 6a3fadc. The issue was found as result of the changes in #16513, however it can be reproduced without as described below.

How to reproduce:

Run the test pytest tests/python/frontend/tflite/test_forward.py -k test_forward_quantized_depthwise_convolution
with an arm_cpu target. Note: Reminder to remove any skip condition that exists in the test currently.


Likely the schedule selection in relay/strategy/arm_cpu.py needs to check compatibility of the output data type before adding the schedule to the strategy.

@lhutton1 lhutton1 added type: bug topi python/tvm/topi needs-triage PRs or issues that need to be investigated by maintainers to find the right assignees to address it labels Feb 7, 2024
@tqchen tqchen closed this as not planned Won't fix, can't repro, duplicate, stale Feb 8, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
needs-triage PRs or issues that need to be investigated by maintainers to find the right assignees to address it topi python/tvm/topi type: bug
Projects
None yet
Development

No branches or pull requests

2 participants