Change fallback operator 2 + update min shape

pytorch · Jun 24, 2024 · c692be6 · c692be6
1 parent 0d8ae44
commit c692be6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/py/dynamo/models/test_dyn_models.py b/tests/py/dynamo/models/test_dyn_models.py
@@ -337,12 +337,12 @@ def forward(self, x):
         "ir": ir,
         "pass_through_build_failures": True,
         "min_block_size": 1,
-        "torch_executed_ops": {"torch.ops.aten.reshape.default"},
+        "torch_executed_ops": {"torch.ops.aten.add.Tensor"},
     }
 
     # Compile the model
     if ir == "torch_compile":
-        torch._dynamo.mark_dynamic(input_bs4, 0, min=2, max=1024)
+        torch._dynamo.mark_dynamic(input_bs4, 0, min=4, max=1024)
         trt_model = torch.compile(model, backend="tensorrt", options=compile_spec)
         trt_model(input_bs4)
     elif ir == "dynamo":