Skip to content

fix(dynamo): handle dynamic scatter and index updates - #4631

Open
narendasan wants to merge 1 commit into
narendasan/fix-truncate-double-metadatafrom
narendasan/fix-dynamic-scatter-index-put
Open

fix(dynamo): handle dynamic scatter and index updates#4631
narendasan wants to merge 1 commit into
narendasan/fix-truncate-double-metadatafrom
narendasan/fix-dynamic-scatter-index-put

Conversation

@narendasan

Copy link
Copy Markdown
Collaborator

Summary

  • skip scatter_add decomposition when its extent is data-dependent so supported dynamic graphs remain compilable
  • normalize index_put update dtype to the destination tensor before TensorRT scatter
  • represent None-index expansion with dynamic arange/broadcast logic instead of one index per statically known element
  • make static arange conversion safe under FakeTensorMode and clamp inferred dynamic profile minima

Issues

Closes #4602
Closes #4609
Closes #4621

Tests

Stack

5 of 7. Base: #4630. Follow-up: narendasan/fix-converter-profile-edge-cases.

@github-actions github-actions Bot added component: tests Issues re: Tests component: lowering Issues re: The lowering / preprocessing passes component: conversion Issues re: Conversion stage component: core Issues re: The core compiler component: converters Issues re: Specific op converters component: api [Python] Issues re: Python API component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: torch_compile labels Aug 28, 2026
@meta-cla meta-cla Bot added the cla signed label Aug 28, 2026
@github-actions
github-actions Bot requested a review from cehongwang August 28, 2026 03:01
@narendasan
narendasan requested a review from apbose August 28, 2026 03:06
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: conversion Issues re: Conversion stage component: converters Issues re: Specific op converters component: core Issues re: The core compiler component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: lowering Issues re: The lowering / preprocessing passes component: tests Issues re: Tests component: torch_compile

Projects

None yet

1 participant