Skip to content

Commit

Permalink
Replace constraints with dynamic_shapes in deeplearning/aot_induc…
Browse files Browse the repository at this point in the history
…tor test

Summary: `constraints` argument for `torch.export` has been deprecated in favor of the `dynamic_shapes` argument. This PR updates the use of the deprecated API in `deeplearning/aot_inductor/test/test_custom_ops.py`.

Test Plan: buck test mode/dev-nosan fbcode//deeplearning/aot_inductor/test:test_custom_ops -- test_export_extern_fallback_nodes_dynamic_shape

Differential Revision: D52790332
  • Loading branch information
BoyuanFeng authored and facebook-github-bot committed Jan 16, 2024
1 parent 3e8c8ce commit 6c07893
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions torch/_export/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,7 @@
)
from torch.export.dynamic_shapes import (
Constraint,
dims,
dynamic_dim,
_process_constraints,
_process_dynamic_shapes,
Expand Down

0 comments on commit 6c07893

Please sign in to comment.