Angelos Katharopoulos
|
3d5e17e507
|
MLX_SWITCH macros to templates (#2320)
|
2025-07-01 01:33:44 -07:00 |
|
Awni Hannun
|
c552ff2451
|
[CUDA] Fix back-end bugs and enable corresponding tests (#2296)
* Fix some cuda back-end bugs and enable corresponding tests
* more fixes
* enable more tests
* format
|
2025-06-16 08:45:40 -07:00 |
|
Angelos Katharopoulos
|
580776559b
|
RoPE for CUDA (#2293)
* First working CUDA rope
* Fix random
|
2025-06-15 06:08:07 -07:00 |
|
Cheng
|
a4fc671d3e
|
CUDA backend: compile (#2276)
* CUDA backend: compile
* Rename kernels/ to device/
|
2025-06-12 17:08:39 -07:00 |
|
Cheng
|
c9fa68664a
|
CUDA backend: reduce (#2269)
|
2025-06-11 11:22:25 -07:00 |
|
Cheng
|
7ebb2e0193
|
CUDA backend: binary ops (#2259)
|
2025-06-10 06:37:40 -07:00 |
|
Cheng
|
f8bad60609
|
CUDA backend: unary ops (#2158)
|
2025-06-09 06:45:08 -07:00 |
|
Cheng
|
f76ee1ffd2
|
Move some dims utils to common (#2223)
|
2025-05-29 06:48:30 -07:00 |
|