zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
Files
012fb220a19035c2ca28ddcf8b2d204eaaf7e761
mlx/mlx/backend/cuda
History
Anastasiia Filippova 012fb220a1
Some checks failed
Build and Test / Check Lint (push) Has been cancelled
Details
Build and Test / Linux (cpu, aarch64) (push) Has been cancelled
Details
Build and Test / Linux (cpu, x86_64) (push) Has been cancelled
Details
Build and Test / Linux (cuda-12.6, aarch64) (push) Has been cancelled
Details
Build and Test / Linux (cuda-12.9, aarch64) (push) Has been cancelled
Details
Build and Test / Linux (cuda-12.6, x86_64) (push) Has been cancelled
Details
Build and Test / Linux (cuda-12.9, x86_64) (push) Has been cancelled
Details
Build and Test / macOS (14.0) (push) Has been cancelled
Details
Build and Test / macOS (15.0) (push) Has been cancelled
Details
Build and Test / Build Documentation (push) Has been cancelled
Details
Build and Test / Linux Fedora (aarch64) (push) Has been cancelled
Details
Build and Test / Linux Fedora (x86_64) (push) Has been cancelled
Details
fp quantize (#2892)
2025-12-11 06:11:25 -08:00
..
binary
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
conv
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
copy
[CUDA] Faster general copy (#2873)
2025-12-04 21:42:15 -08:00
device
Fp8 conversion (#2686)
2025-10-27 16:35:50 -07:00
gemms
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
quantized
fp quantize (#2892)
2025-12-11 06:11:25 -08:00
reduce
Add a 2-pass col reduce for CUDA (#2863)
2025-12-04 15:53:59 -08:00
steel
fp quantize (#2892)
2025-12-11 06:11:25 -08:00
unary
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
allocator.cpp
Try not to fail when there should be memory available (#2869)
2025-12-07 06:11:00 -08:00
allocator.h
Try not to fail when there should be memory available (#2869)
2025-12-07 06:11:00 -08:00
arange.cu
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
arg_reduce.cu
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
bin2h.cmake
CUDA backend: compile (#2276)
2025-06-12 17:08:39 -07:00
binary_two.cu
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
CMakeLists.txt
Merge build-cuda and build-linux actions (#2783)
2025-11-25 20:06:42 +09:00
compiled.cpp
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
conv.cpp
[CUDA] Migrate conv code to new cuDNN APIs (#2847)
2025-12-02 07:55:43 +09:00
copy.cu
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
cuda_utils.h
[CUDA] Migrate conv code to new cuDNN APIs (#2847)
2025-12-02 07:55:43 +09:00
cuda.cpp
start cuda circle config (#2256)
2025-06-10 21:19:47 -07:00
cuda.h
start cuda circle config (#2256)
2025-06-10 21:19:47 -07:00
cudnn_utils.cpp
[CUDA] Migrate conv code to new cuDNN APIs (#2847)
2025-12-02 07:55:43 +09:00
cudnn_utils.h
[CUDA] Migrate conv code to new cuDNN APIs (#2847)
2025-12-02 07:55:43 +09:00
custom_kernel.cpp
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
device.cpp
Allow events in sub graph to be updatable (#2886)
2025-12-09 12:34:37 -08:00
device.h
[CUDA] Enable more graphs to be updatable (#2883)
2025-12-08 06:18:01 -08:00
distributed.cu
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
eval.cpp
[CUDA] Tune ops per buffer based on device (#2761)
2025-11-16 06:29:49 -08:00
event.cu
[CUDA] Exit on crash and more helpful errors (#2830)
2025-11-24 19:46:03 -08:00
event.h
[CUDA] Make CudaEvent work with multi-device (#2614)
2025-09-27 11:27:17 +09:00
fence.cpp
[CUDA] Reduce use of managed memory (#2725)
2025-11-05 16:05:23 -08:00
indexing.cpp
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
jit_module.cpp
use arch specific targets when possible (#2771)
2025-11-14 20:04:18 -08:00
jit_module.h
[CUDA] Reduce use of managed memory (#2725)
2025-11-05 16:05:23 -08:00
kernel_utils.cu
fix for max block dim (#2631)
2025-09-29 08:59:25 -07:00
kernel_utils.cuh
[CUDA] Reduce use of managed memory (#2725)
2025-11-05 16:05:23 -08:00
layer_norm.cu
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
load.cpp
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
logsumexp.cu
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
lru_cache.h
[CUDA] cuDNN forward attention (#2743)
2025-11-14 09:23:56 +09:00
matmul.cpp
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
no_cuda.cpp
Custom cuda kernel (#2517)
2025-08-20 17:20:22 -07:00
primitives.cpp
Add Masked Scatter (#2663)
2025-11-19 14:53:32 -08:00
random.cu
[CUDA] Partly fix random for large sizes (#2798)
2025-11-20 07:27:50 -08:00
reduce.cu
faster rms norm (#2433)
2025-07-29 13:12:00 -07:00
rms_norm.cu
[CUDA] Faster rms norm for small dimension (#2838)
2025-11-26 15:10:41 -08:00
rope.cu
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
scaled_dot_product_attention.cpp
[CUDA] Migrate conv code to new cuDNN APIs (#2847)
2025-12-02 07:55:43 +09:00
scaled_dot_product_attention.cu
[CUDA] cuDNN backward attention (#2762)
2025-11-19 08:13:50 +09:00
scan.cu
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
slicing.cpp
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
softmax.cu
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
sort.cu
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
ternary.cu
Fix building with CUDA < 12.8 (#2782)
2025-11-18 12:55:19 +09:00
utils.cpp
[CUDA] Enable more graphs to be updatable (#2883)
2025-12-08 06:18:01 -08:00
utils.h
[CUDA] Migrate conv code to new cuDNN APIs (#2847)
2025-12-02 07:55:43 +09:00
vector_types.cuh
fp quantize (#2892)
2025-12-11 06:11:25 -08:00
worker.cpp
[CUDA] Check CUDA error in synchronize (#2757)
2025-11-14 07:10:23 +09:00
worker.h
[CUDA] Make CudaEvent work with multi-device (#2614)
2025-09-27 11:27:17 +09:00