Skip to content

Fix input/output scalar merging in compile - #4658

Merged
zcbenz merged 2 commits into
ml-explore:mainfrom
michalk8:fix-compile-scalar-output
Oct 9, 2026
Merged

zcbenz merged 2 commits into
ml-explore:mainfrom
michalk8:fix-compile-scalar-output

Conversation

@michalk8

@michalk8 michalk8 commented Oct 8, 2026 •

Copy link
Copy Markdown
Collaborator
  • ☑️ I understand it is strictly prohibited to use AI to write PR description
  • AI usage disclosure: Tests

Fixes mx.compile scalar merging when retuning a scalar output (was throwing E IndexError: unordered_map::at: key not found). Now, stuff like lambda x: (x + 1.0, mx.array(1.0) compiles just fine.

@zcbenz
zcbenz force-pushed the fix-compile-scalar-output branch from 0421a31 to ca0401f Compare October 9, 2026 10:11
@zcbenz
zcbenz merged commit 84a72e3 into ml-explore:main Oct 9, 2026
30 checks passed
@michalk8
michalk8 deleted the fix-compile-scalar-output branch October 9, 2026 14:15
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