Code

Add the MLX merge sort kernels#2751

Code

Merged

LaurentMazare merged 11 commits intomainhuggingface/candle:mainfrom mlx-sorthuggingface/candle:mlx-sortCopy head branch name to clipboardJan 28, 2025

+1,426-213

Commits

Commits on Jan 26, 2025

• Add some metal sort kernels imported from MLX.

  LaurentMazare

  committed
  82c7998

  Copy full SHA for 82c7998

• Add another test.

  LaurentMazare

  committed
  96ee9d8

  Copy full SHA for 96ee9d8

Commits on Jan 27, 2025

• Start adding the multiblock version.

  LaurentMazare

  committed
  1846f6b

  Copy full SHA for 1846f6b

• Merge branch 'main' into mlx-sort

  LaurentMazare

  committed
  373cc79

  Copy full SHA for 373cc79

• Proper kernel names.

  LaurentMazare

  committed
  38076ce

  Copy full SHA for 38076ce

Commits on Jan 28, 2025

• Merge remote-tracking branch 'origin/main' into mlx-sort

  LaurentMazare

  committed
  08bce96

  Copy full SHA for 08bce96

• Split out the main metal file.

  LaurentMazare

  committed
  8c15767

  Copy full SHA for 8c15767

• Multi-block sort.

  LaurentMazare

  committed
  e7da916

  Copy full SHA for e7da916

• More sorting.

  LaurentMazare

  committed
  9966f5c

  Copy full SHA for 9966f5c

• DType parametrization.

  LaurentMazare

  committed
  20ccdf7

  Copy full SHA for 20ccdf7

• Add a larger test.

  LaurentMazare

  committed
  8c2f7af

  Copy full SHA for 8c2f7af