Skip to content

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

Description

@lhutton1

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.

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    needs-triagePRs or issues that need to be investigated by maintainers to find the right assignees to address ittopipython/tvm/topitype: bug

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions