ONNX CI workflow is broken #5971
Description
Since the 5th of May our CI workflow for ONNX is broken (commit 970ba35). Looking at the warnings emitted by the failing tests
WARNING: The shape inference of prim::Constant type is missing, so it may result in wrong shape inference for the exported graph. Please consider adding it in symbolic function.
[...]
AssertionError: The values for attribute 'shape' do not match: torch.Size([1, 4]) != torch.Size([0, 4]).
Two models are affected faster_rcnn and mask_rcnn. To reproduce run:
pytest test/test_onnx.py -k "test_faster_rcnn"
pytest test/test_onnx.py -k "test_mask_rcnn"
I believe a recent patch to primtorch might be the offender here. cc @neginraoof @seemethere @mruberry