ROCm / AMDMIGraphX

AMD's graph optimization engine.
https://rocm.docs.amd.com/projects/AMDMIGraphX/en/latest/
MIT License
185 stars 86 forks source link

ONNX files for Opset 8 fp16_tiny_yolov gets errors #444

Closed mvermeulen closed 4 years ago

mvermeulen commented 4 years ago

Running this ONNX file with the MIGraphX driver results in the following errors

terminate called after throwing an instance of 'migraphx::version_1::exception'
  what():  /home/mev/source/AMDMIGraphX/src/include/migraphx/check_shapes.hpp:84
: Types do not match

Not 100% certain these are the same model, but fp32 Opset 8 version of tiny_yolov2 works without error. I've attached a dump of ONNX files for both fp16 and fp32 models yolov_fp32.txt yolov_fp16.txt

mvermeulen commented 4 years ago

Note: Both the fp32 and fp16 versions are created from WinMLTools. These are the only models in the test set that come from this provider.

scxiao commented 4 years ago

migraphx considers all attribute as float type, but input of the model is fp16, so there is a typemismatch.

scxiao commented 4 years ago

created a PR for the fix

mvermeulen commented 4 years ago

Implemented