Change _dynamo.export to be export(f)(*args, **kwargs) (#106109)

Summary:
Signed-off-by: Edward Z. Yang <[email protected]>

X-link: pytorch/pytorch#106109
Approved by: https://github.com/voznesenskym

bypass-github-export-checks

Reviewed By: PaliC

Differential Revision: D47860637

Pulled By: ezyang

fbshipit-source-id: 3933e3ffc43929f27ca4d3dc44a9bc822e8bc8db

Author: ezyang

exir/tracer.py

@@ -674,14 +674,15 @@ def dynamo_trace(
             # functionalization
             return torchdynamo.export(
                 f,
-                *copy.deepcopy(args),
                 aten_graph=aten_graph,
                 tracing_mode=tracing_mode,
                 assume_static_by_default=dynamo_config.assume_static_by_default,
                 decomposition_table=_default_decomposition_table(_use_old_decomp_table)
                 if aten_graph
                 else None,
                 constraints=constraints,
+            )(
+                *copy.deepcopy(args),
             )
         except torchdynamo.exc.Unsupported as exc:
             raise ExportError(