Skip to content

Fix/split data size span - #4540

Open
michalk8 wants to merge 5 commits into
ml-explore:mainfrom
michalk8:fix/split-data-size-span
Open

michalk8 wants to merge 5 commits into
ml-explore:mainfrom
michalk8:fix/split-data-size-span

Conversation

@michalk8

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

Fix mx.split incorrectly computing its data size. Discovered in ml-explore/mlx-lm#1896 when computing RoPE on non-contiguous slices.

@michalk8
michalk8 requested a review from zcbenz September 21, 2026 12:46
@michalk8
michalk8 requested a review from nastya236 September 21, 2026 12:52
@nastya236 nastya236 added the await verification This pull request is non-trivial and requires a human expert to verify its correctness. label Sep 22, 2026

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

await verification This pull request is non-trivial and requires a human expert to verify its correctness.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants