Skip to content

Fix (proxy): preserve training state after tensor_quant re-init#1419

Merged
Giuseppe5 merged 8 commits intoXilinx:devfrom
Giuseppe5:rotate_fix
Dec 2, 2025
Merged

Fix (proxy): preserve training state after tensor_quant re-init#1419
Giuseppe5 merged 8 commits intoXilinx:devfrom
Giuseppe5:rotate_fix

Conversation

@Giuseppe5
Copy link
Collaborator

Reason for this PR

When merging rotations, the state and device of the quantizers might be lost.

Changes Made in this PR

Preserve and restore device and training state.

Testing Summary

TBD

@Giuseppe5 Giuseppe5 requested a review from pablomlago November 28, 2025 16:09
Copy link
Collaborator

@pablomlago pablomlago left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Is it possible to have some test to verify that the fix works as intended?

@Giuseppe5 Giuseppe5 changed the title Fix (equalize/rotations): preserve state and device after rot fusion Fix (proxy): preserve training state after tensor_quant re-init Nov 29, 2025
@pablomlago
Copy link
Collaborator

pablomlago commented Dec 1, 2025

LGTM (assuming tests pass)

@Giuseppe5 Giuseppe5 requested a review from pablomlago December 1, 2025 12:10
@Giuseppe5 Giuseppe5 merged commit 216101e into Xilinx:dev Dec 2, 2025
484 of 486 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants