Skip to content

metal : faster argsort #17315

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Merged
merged 2 commits into from
Nov 17, 2025
Merged

metal : faster argsort #17315

merged 2 commits into from
Nov 17, 2025
+87 −41

Conversation

@ggerganov
Copy link
Member

@ggerganov ggerganov commented Nov 17, 2025

cont #17247

  • One binary-search per thread
  • Linear merge for the rest of the elements
  • Keep merge fronts in registers to reduce global reads

@github-actions github-actions bot added ggml changes relating to the ggml tensor library for machine learning Apple Metal https://en.wikipedia.org/wiki/Metal_(API) labels Nov 17, 2025
@DajanaV DajanaV mentioned this pull request Nov 17, 2025
Open
@ggerganov ggerganov merged commit 3347e6d into master Nov 17, 2025
67 of 72 checks passed
@ggerganov ggerganov deleted the gg/metal-argsort-opt branch November 17, 2025 09:51
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

No reviews

Assignees

No one assigned

Labels

Apple Metal https://en.wikipedia.org/wiki/Metal_(API) ggml changes relating to the ggml tensor library for machine learning

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@ggerganov