Replace export_for_training with export

Summary: torch.export.export_for_training will be deprecated

Differential Revision: D78751416

Author: ethansfng

backends/cadence/aot/compiler.py

@@ -59,7 +59,7 @@ def trace(
     dump_graphs: bool = False,
 ) -> ExportedProgram:
     """
-    Trace the model with export_for_training and return an ExportedProgram.
+    Trace the model with export and return an ExportedProgram.
     """
 
     # Make the model inference mode by calling model.eval()
@@ -83,7 +83,7 @@ def trace(
     remove_decompositions(decomp_table, ops_to_keep)
 
     # Export with dynamo
-    program = torch.export.export_for_training(
+    program = torch.export.export(
         model, inputs, strict=True
     ).run_decompositions(decomp_table)