diff --git a/doc/releases/changelog-dev.md b/doc/releases/changelog-dev.md index 1d3e52359..6b176e0f1 100644 --- a/doc/releases/changelog-dev.md +++ b/doc/releases/changelog-dev.md @@ -13,6 +13,7 @@ * Fixes the translation of plxpr control flow for edge cases where the `consts` were being reordered. [(#2128)](https://github.com/PennyLaneAI/catalyst/pull/2128) + [(#2133)](https://github.com/PennyLaneAI/catalyst/pull/2133)

Internal changes ⚙️

@@ -21,3 +22,6 @@

Contributors ✍️

This release contains contributions from (in alphabetical order): + +Christina Lee, +Paul Haochen Wang. diff --git a/frontend/catalyst/from_plxpr/control_flow.py b/frontend/catalyst/from_plxpr/control_flow.py index 663a55436..37de2ed5d 100644 --- a/frontend/catalyst/from_plxpr/control_flow.py +++ b/frontend/catalyst/from_plxpr/control_flow.py @@ -221,7 +221,7 @@ def handle_for_loop( outvals = for_p.bind( *for_loop_invals, body_jaxpr=converted_closed_jaxpr_branch, - body_nconsts=len(consts), + body_nconsts=len(new_consts), apply_reverse_transform=apply_reverse_transform, nimplicit=0, preserve_dimensions=True,