Skip to content

Skip adjacency partitioning when the graph is fully supported - #4527

Closed
micwill755 wants to merge 1 commit into
pytorch:mainfrom
micwill755:partition-short-circuit
Closed

Skip adjacency partitioning when the graph is fully supported#4527
micwill755 wants to merge 1 commit into
pytorch:mainfrom
micwill755:partition-short-circuit

Conversation

@micwill755

@micwill755 micwill755 commented Aug 19, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • Skip adjacency partitioning when require_full_compilation already has full converter coverage, wrapping the graph as a single TRT block instead of repeating AccNodesFinder, fusion, and split.

This is PR 1/3 in a compile-startup stack:

  1. This PR
  2. Skip installing large aliased constant-folded weights #4528
  3. Batch cheap FX cleanups and defer post_lowering recompile #4529

Test plan

  • Compile a fully supported model with require_full_compilation=True and confirm one _run_on_acc_* submodule
  • Compile a mixed-support graph and confirm adjacency partitioning still runs
  • Flux / VLA compile still succeeds with this path enabled

When require_full_compilation already has full converter coverage, wrap the graph as one TRT block instead of repeating AccNodesFinder, fusion, and adjacency split.
@meta-cla meta-cla Bot added the cla signed label Aug 19, 2026
@github-actions github-actions Bot added component: core Issues re: The core compiler component: api [Python] Issues re: Python API component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths labels Aug 19, 2026
@github-actions
github-actions Bot requested a review from narendasan August 19, 2026 21:50
@micwill755

Copy link
Copy Markdown
Collaborator Author

Superseded by the real GitHub stack (stack #4533): #4530 -> #4531 -> #4532. These fork branches could not be stacked because GitHub stacked PRs do not support pull requests from forks.

@micwill755 micwill755 closed this Aug 19, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

cla signed component: api [Python] Issues re: Python API component: core Issues re: The core compiler component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant