I'm trying to compile yolov8s on AI 100 using the SDK. However, it seems it's failing because it has to evenly split a tensor from the very last Split operator. Assuming the input is 1x3x640x640, I see the very last Split operator is trying to split a tensor under 1x144x8400 into a tensor under 1x80x8400 and a tensor under 1x64x8400 from onnx. However, the SDK seems to be doing two 1x72x8400 tensors and the compilation is failing.
Has anybody seen similar issues before? Thanks ahead!
Hello cchangl,
So far ,aic100 compiler can only support static shape as input. If you compiled your yolov8s model with aic100 compiler successfully, you can view its
input and output shape information via below command.
/opt/qti-aic/tools/qaic-qpc validate -i <your folder which contain the programqpc.bin>