Migrate devtools tests and clean up remaining executorch/ CaptureConfig refs#18135
Migrate devtools tests and clean up remaining executorch/ CaptureConfig refs#18135JacobSzwejbka wants to merge 3 commits intopytorch:mainfrom
Conversation
Summary: Migrate test_pass_infra, test_debug_handle_map, test_delegate_map_builder, test_backends_nested, and hta_partitioner_demo to use the torch.export.export() + to_edge() flow instead of the deprecated exir.capture() API. Key changes: - Replace exir.capture(model, inputs, CaptureConfig()).to_edge() with to_edge(export(model, inputs, strict=True)) - Wrap plain functions in nn.Module for torch.export compatibility - Use dynamic debug handle extraction instead of hardcoded values in test_delegate_map_builder (handle numbering changed) - Collapse lifted/unlifted pattern variants in hta_partitioner_demo since torch.export always produces lifted graphs Differential Revision: D95605454
Summary: Delete capture_graph_for_xnnpack() and get_xnnpack_capture_config() which were only used in test files and relied on the deprecated exir.capture API. Migrate test_xnnpack_utils.py to use inline to_edge(export(...)) calls. Remove the deprecated exports from xnnpack/__init__.py. Differential Revision: D95605468
…ig refs Summary: Migrate etrecord_test, exported_op_graph_test, and size_analysis_tool_test from exir.capture to torch.export + to_edge. Remove dead CaptureConfig imports/usage from end2end tests, export_program, and qualcomm utils (delete qnn_capture_config() which returned a CaptureConfig). Differential Revision: D95605485
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/18135
Note: Links to docs will display an error until the docs builds have been completed. ❌ 1 Awaiting Approval, 5 New Failures, 5 Pending, 3 Unrelated FailuresAs of commit 96ac69c with merge base eaf0d65 ( NEW FAILURES - The following jobs have failed:
BROKEN TRUNK - The following jobs 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:
Migrate etrecord_test, exported_op_graph_test, and size_analysis_tool_test
from exir.capture to torch.export + to_edge. Remove dead CaptureConfig
imports/usage from end2end tests, export_program, and qualcomm utils
(delete qnn_capture_config() which returned a CaptureConfig).
Differential Revision: D95605485