Change fallback operator

pytorch · Jun 24, 2024 · 0d8ae44 · 0d8ae44
1 parent 54047df
commit 0d8ae44
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/py/dynamo/models/test_dyn_models.py b/tests/py/dynamo/models/test_dyn_models.py
@@ -337,7 +337,7 @@ def forward(self, x):
         "ir": ir,
         "pass_through_build_failures": True,
         "min_block_size": 1,
-        "torch_executed_ops": {"torch.ops.aten.add.Tensor"},
+        "torch_executed_ops": {"torch.ops.aten.reshape.default"},
     }
 
     # Compile the model