mlx/mlx/backend
Cheng 8347575ba1
[CUDA] Implement Scan kernel (#2347)
* Contiguous scan

* Strided scan

* Enable tests

* Fix failing logaddexp test

* Use cexpf in Metal
2025-07-10 16:54:12 -07:00
..
common [CUDA] Switch to CUDA graphs (#2317) 2025-07-02 15:59:13 -07:00
cpu Align mlx::core::min op nan propagation with NumPy (#2346) 2025-07-10 06:20:43 -07:00
cuda [CUDA] Implement Scan kernel (#2347) 2025-07-10 16:54:12 -07:00
gpu Make sliceUpdate general (#2282) 2025-06-12 16:48:54 -07:00
metal [CUDA] Implement Scan kernel (#2347) 2025-07-10 16:54:12 -07:00
no_cpu MoE backward improvements (#2335) 2025-07-07 17:59:53 -07:00
no_gpu MoE backward improvements (#2335) 2025-07-07 17:59:53 -07:00