diff --git a/backends/cadence/aot/compiler.py b/backends/cadence/aot/compiler.py index 560b625e4c0..844e35e3d26 100644 --- a/backends/cadence/aot/compiler.py +++ b/backends/cadence/aot/compiler.py @@ -88,7 +88,7 @@ def trace( # Export with dynamo program = torch.export.export_for_training( - model, inputs, strict=True + model, inputs ).run_decompositions(decomp_table) if dump_graphs: