modular / modular/modular

[Feature Request] Add gpu sort to the standard library

Open
#6,709 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

enhancement mojo Needs Triage
Dominant language
Mojo
Stars
29.8k
Forks
3.2k
PR merge metrics
No merged PRs in 30d

Description

Review Mojo's priorities
What is your request?

I have a a few gpu sorting algorithms implemented in mojo : Bitonic Sort, Radix Sort, Onesweep.

Is it something that mojo would consider incorporating in the standard library ?

the implementation is in this repo: https://github.com/jgsimard/bajo/tree/main/bajo/sort/gpu

Performance is very close to CUB !

Image

the image is computed from benchmarks here

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by reviewing the Mojo roadmap and priorities linked in the issue, then inspect the proposed implementations in bajo/sort/gpu and the benchmarks in bench/sort. Determine whether GPU sorting belongs in the standard library and what API and algorithm scope would be appropriate. Done means an agreed library design and an accepted implementation path backed by the provided benchmark results.

Written by the indexing model from the issue text.

Assessment

Domain
tooling
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Quiet
Clarity
Mostly clear
Newbie friendliness
38/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.