From 66a8559edce40951187b9756b71f5003948b8b6b Mon Sep 17 00:00:00 2001 From: paul0403 Date: Mon, 20 Oct 2025 16:49:28 -0400 Subject: [PATCH 1/2] Small follow-up on #2128 --- frontend/catalyst/from_plxpr/control_flow.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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, From c32b8451e8eb0b677690775048bf9276d6bbff40 Mon Sep 17 00:00:00 2001 From: paul0403 Date: Mon, 20 Oct 2025 16:56:07 -0400 Subject: [PATCH 2/2] changelog --- doc/releases/changelog-dev.md | 4 ++++ 1 file changed, 4 insertions(+) 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.