Skip to content

Commit c2ba8e6

Browse files
committed
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
1 parent 2c78f43 commit c2ba8e6

File tree

3 files changed

+1
-4
lines changed

3 files changed

+1
-4
lines changed

.github/ci_commit_pins/vision.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
fc838add3235d60feaea2251c1979fd253b26a77
1+
452caaaf9a46f36390f00e7f66066c3b4edfaf22

benchmarks/dynamo/common.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,6 @@ class CI(NamedTuple):
102102
"hf_BigBird", # fp64_OOM
103103
"hf_T5_base", # fp64_OOM
104104
"llama", # Accuracy failed: allclose not within tol=0.001
105-
"vision_maskrcnn", # shape mismatch in accuracy check!
106105
# Huggingface
107106
"XGLMForCausalLM", # OOM
108107
# TIMM
@@ -259,7 +258,6 @@ class CI(NamedTuple):
259258
*CI_SKIP[CI("aot_eager", training=False)],
260259
"cm3leon_generate", # Could not validate constraint UnspecConstraint
261260
"hf_T5_generate", # Could not validate constraint UnspecConstraint
262-
"vision_maskrcnn", # nothing in example_inputs had a dim with 4
263261
]
264262

265263
CI_SKIP[CI("aot_eager", training=True, dynamic=True)] = [

benchmarks/dynamo/torchbench.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,6 @@ def setup_torchbench_cwd():
8181
"hf_T5_generate", # OOMs
8282
"cm3leon_generate", # model is CUDA only
8383
"nanogpt_generate", # timeout
84-
"vision_maskrcnn", # The size of tensor a (36) must match the size of tensor b (34)
8584
}
8685

8786
SKIP_FOR_CUDA = {

0 commit comments

Comments
 (0)