Skip to content

fix(dynamo): remap engine output symbols during fake execution - #4627

Open
narendasan wants to merge 1 commit into
mainfrom
narendasan/fix-engine-symbol-remap
Open

fix(dynamo): remap engine output symbols during fake execution#4627
narendasan wants to merge 1 commit into
mainfrom
narendasan/fix-engine-symbol-remap

Conversation

@narendasan

Copy link
Copy Markdown
Collaborator

Summary

  • remap serialized engine output expressions into the active runtime ShapeEnv instead of reusing symbols across traces
  • solve composite input expressions against runtime dimensions
  • preserve size-like constraints on fresh data-dependent output symbols
  • support symbolic outputs when statically shaped inputs do not provide a ShapeEnv

Issues

Closes #4618
Closes #4619
Closes #4622

Tests

Stack

1 of 7. Base: main. Follow-up: narendasan/fix-engine-scalar-io.

Remap serialized compile-time shape expressions into the active runtime ShapeEnv instead of reusing symbol names across namespaces. Fresh data-dependent output symbols retain size-like constraints, and composite input expressions are solved against runtime dimensions.

Fixes #4618
Fixes #4619
Fixes #4622

Tests:
- pytest tests/py/dynamo/models/test_meta_kernel_shape_inference.py -n0 (10 passed)
- original issue reproductions for #4618, #4619, and #4622
@meta-cla meta-cla Bot added the cla signed label Aug 28, 2026
@github-actions github-actions Bot added component: tests Issues re: Tests component: core Issues re: The core compiler component: api [Python] Issues re: Python API component: runtime component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths labels Aug 28, 2026
@github-actions
github-actions Bot requested a review from lanluo-nvidia August 28, 2026 02:58
@narendasan
narendasan requested a review from apbose August 28, 2026 03:05
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 component: runtime component: tests Issues re: Tests

Projects

None yet

1 participant