caffe2/benchmarks/dynamo (#143686)

Summary: Pull Request resolved: pytorch/pytorch#143686 Reviewed By: avikchaudhuri, tugsbayasgalan Differential Revision: D67529443 fbshipit-source-id: fbc822c7eb0da7fc95bd1d4e216476cc8adcaaa2
pytorch · Dec 21, 2024 · 0002f94 · 0002f94
1 parent 2bd7552
commit 0002f94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/userbenchmark/dynamo/dynamobench/common.py b/userbenchmark/dynamo/dynamobench/common.py
@@ -1622,7 +1622,7 @@ def export(model, example_inputs):
     )
 
     ep = torch.export.export(
-        model, example_args, example_kwargs, dynamic_shapes=dynamic_shapes
+        model, example_args, example_kwargs, dynamic_shapes=dynamic_shapes, strict=True
     )
 
     def opt_export(_, example_inputs):