migraphx-benchmark / AMDMIGraphX

AMD's graph optimization engine.
https://rocmsoftwareplatform.github.io/AMDMIGraphX/doc/html/
MIT License
0 stars 1 forks source link

Parse fail with SSD-MobileNetV1 models #149

Open attila-dusnoki-htec opened 11 months ago

attila-dusnoki-htec commented 11 months ago
SSD-MobileNetV1 ssd_mobilenet_v1_10.onnx 10 reshape: Wrong number of elements for reshape https://github.com/onnx/models/blob/main/vision/object_detection_segmentation/ssd-mobilenetv1/model/ssd_mobilenet_v1_10.onnx
SSD-MobileNetV1 ssd_mobilenet_v1_12.onnx 12 reshape: Wrong number of elements for reshape https://github.com/onnx/models/blob/main/vision/object_detection_segmentation/ssd-mobilenetv1/model/ssd_mobilenet_v1_12.onnx
SSD-MobileNetV1 ssd_mobilenet_v1_13-qdq.onnx 13 reshape: Wrong number of elements for reshape https://github.com/onnx/models/blob/main/vision/object_detection_segmentation/ssd-mobilenetv1/model/ssd_mobilenet_v1_13-qdq.onnx

During the above listed reshape issue, we identified that the problem was with the max loop iterations. This will get fixed in https://github.com/ROCmSoftwarePlatform/AMDMIGraphX/pull/2361.

With that applied, it will continue the parse and will fail with

It will result in /code/AMDMIGraphX/src/include/migraphx/check_shapes.hpp:119: has: slice: Wrong number of arguments: expected 1, 3, 4 but given 2 for parsing slice.

It fails at this node:

arg0: squeeze[axes={0}](gather[axis=0]) -> float_type, {1917, 90}, {90, 1}
arg1: @literal{0, 0} -> int32_type, {2}, {1}
arg2: add(@literal{0, 0}, add) -> int32_type, {2}, {1}

where arg0 is operator: Squeeze

Loop_1114_loop:@1312 = squeeze[axes={0}](Loop_1114_loop:@1311) -> float_type, {1917, 90}, {90, 1}, target_id=0
attila-dusnoki-htec commented 11 months ago

This issue seems to be related: https://github.com/ROCmSoftwarePlatform/AMDMIGraphX/issues/2349