Add pass to remove non-contiguous dim order from a graph.#21057
Add pass to remove non-contiguous dim order from a graph.#21057MartinPavella wants to merge 1 commit into
Conversation
The pass preserves the dim order of the inputs, but internally the graph will use only the contiguous dim order.
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/21057
Note: Links to docs will display an error until the docs builds have been completed. ❌ 1 New Failure, 1 Unrelated FailureAs of commit 16832a4 with merge base a19d1ba ( NEW FAILURE - The following job has failed:
BROKEN TRUNK - The following job failed but were present on the merge base:👉 Rebase onto the `viable/strict` branch to avoid these failures
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
|
This PR needs a
|
Summary
The second (and last) part of #20094. The pass preserves the dim order of the inputs, but internally the graph will use only the contiguous dim order.
For models exported with channels last example inputs,
dim_order_cloneoperators will be inserted to make the inputs contiguous internally. Thesedim_order_cloneops can then be targeted by a solution from #20095.Test plan
pytest backends/transforms/test/test_enforce_contiguous_dim_order.py