Fix shape mismatch in vision_maskrcnn

The bulk of the heavy lifting is happening in
pytorch/vision#7592

Signed-off-by: Edward Z. Yang <ezyangmeta.com>

ghstack-source-id: d8f8c95
Pull Request resolved: #101477

Author: ezyang

.github/ci_commit_pins/vision.txt

@@ -1 +1 @@
-fc838add3235d60feaea2251c1979fd253b26a77
+452caaaf9a46f36390f00e7f66066c3b4edfaf22

benchmarks/dynamo/common.py

@@ -102,7 +102,6 @@ class CI(NamedTuple):
     "hf_BigBird",  # fp64_OOM
     "hf_T5_base",  # fp64_OOM
     "llama",  # Accuracy failed: allclose not within tol=0.001
-    "vision_maskrcnn",  # shape mismatch in accuracy check!
     # Huggingface
     "XGLMForCausalLM",  # OOM
     # TIMM
@@ -259,7 +258,6 @@ class CI(NamedTuple):
     *CI_SKIP[CI("aot_eager", training=False)],
     "cm3leon_generate",  # Could not validate constraint UnspecConstraint
     "hf_T5_generate",  # Could not validate constraint UnspecConstraint
-    "vision_maskrcnn",  # nothing in example_inputs had a dim with 4
 ]
 
 CI_SKIP[CI("aot_eager", training=True, dynamic=True)] = [

benchmarks/dynamo/torchbench.py

@@ -81,7 +81,6 @@ def setup_torchbench_cwd():
     "hf_T5_generate",  # OOMs
     "cm3leon_generate",  # model is CUDA only
     "nanogpt_generate",  # timeout
-    "vision_maskrcnn",  # The size of tensor a (36) must match the size of tensor b (34)
 }
 
 SKIP_FOR_CUDA = {