Skip to content

fix(dynamo): make truncate_double metadata match engine precision - #4630

Open
narendasan wants to merge 1 commit into
narendasan/fix-device-placementfrom
narendasan/fix-truncate-double-metadata
Open

fix(dynamo): make truncate_double metadata match engine precision#4630
narendasan wants to merge 1 commit into
narendasan/fix-device-placementfrom
narendasan/fix-truncate-double-metadata

Conversation

@narendasan

Copy link
Copy Markdown
Collaborator

Summary

  • use partition FX metadata instead of executing TensorRT subgraphs during compilation
  • record truncated float32 engine bindings accurately
  • preserve the explicit float64 restoring cast and its output metadata
  • handle non-tensor outputs and CPU constants without probing the engine

Issues

Closes #4604
Closes #4623
Closes #4624

Tests

Stack

4 of 7. Base: #4629. Follow-up: narendasan/fix-dynamic-scatter-index-put.

Fixes #4604.
Fixes #4623.
Fixes #4624.

Use partition FX metadata instead of executing TensorRT subgraphs during compilation. Record truncated float32 engine bindings accurately while preserving the explicit float64 restoring cast and its metadata.
@github-actions github-actions Bot added component: tests Issues re: Tests component: conversion Issues re: Conversion stage 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 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:00
@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: core Issues re: The core compiler component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths component: tests Issues re: Tests

Projects

None yet

1 participant