Skip to content

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

@lhutton1

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.

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