Skip to content

quantized pytorch model convert bug in upsample_nearest2d #7645

@eleflea

Description

@eleflea

When I convert a QNN with aten::upsample_nearest2d op, I encountered a error with:

File "/home/eleflea/.local/lib/python3.7/site-packages/tvm-0.8.dev704+g3a0e3a5bb-py3.7-linux-x86_64.egg/tvm/relay/frontend/pytorch.py", line 1749, in upsample
assert isinstance(inputs[-2], float)

After some research I found that for some quantized op, tvm add qparams manually to help convert. But for aten::upsample_nearest2d, it seems to be missing.

So I add aten::upsample_nearest2d just above aten::upsample_bilinear2d at here. And it works.

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type
No fields configured for issues without a type.

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions