Skip to content

Commit

Permalink
specify output type as torch backend contract form
Browse files Browse the repository at this point in the history
  • Loading branch information
sjain-stanford committed Jul 15, 2024
1 parent 93016be commit 235ff6d
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions tools/aot/torch_exporter_harness.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,6 +48,11 @@ def main():
*loader_result.inputs, # unpack list of input tensors
dynamic_shapes=loader_result.dynamic_shapes,
import_symbolic_shape_expressions=True,
# This is the Torch dialect imported from Dynamo/FX export and run
# through `torchdynamo-export-to-torch-backend-pipeline` (which
# runs `ReduceOpVariantsPass` and `DecomposeComplexOpsPass`) to
# get it in a backend compliant form (aka torch backend contract).
output_type="torch",
func_name=loader_result.func_name,
)

Expand Down

0 comments on commit 235ff6d

Please sign in to comment.