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 1 Packages Projects Releases Wiki Activity
Files
6e762fe2e2c07886fe3fd47c6f22df1a8cdd8c66
mlx/mlx/backend/metal
History
Cheng 704fd1ae28
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
Nightly Build / build_linux_release (3.10) (push) Has been cancelled
Details
Nightly Build / build_linux_release (3.14) (push) Has been cancelled
Details
Nightly Build / build_linux_with_tests (3.11, ubuntu-22.04) (push) Has been cancelled
Details
Nightly Build / build_linux_with_tests (3.11, ubuntu-22.04-arm) (push) Has been cancelled
Details
Nightly Build / build_linux_with_tests (3.12, ubuntu-22.04) (push) Has been cancelled
Details
Nightly Build / build_linux_with_tests (3.12, ubuntu-22.04-arm) (push) Has been cancelled
Details
Nightly Build / build_linux_with_tests (3.13, ubuntu-22.04) (push) Has been cancelled
Details
Nightly Build / build_linux_with_tests (3.13, ubuntu-22.04-arm) (push) Has been cancelled
Details
Nightly Build / build_linux_with_tests (3.14, ubuntu-22.04) (push) Has been cancelled
Details
Nightly Build / build_linux_with_tests (3.14, ubuntu-22.04-arm) (push) Has been cancelled
Details
Nightly Build / build_mac_release (3.10) (push) Has been cancelled
Details
Nightly Build / build_mac_release (3.13) (push) Has been cancelled
Details
Nightly Build / build_cuda_release (push) Has been cancelled
Details
[CUDA] Support array mask in SDPA (#2822)
2025-11-26 11:08:58 +09:00
..
jit
Add Masked Scatter (#2663)
2025-11-19 14:53:32 -08:00
kernels
Centralize NAX condition (#2811)
2025-11-21 13:28:15 -08:00
allocator.cpp
fix warnings showing up with -Wall (#2692)
2025-10-24 11:43:35 -07:00
allocator.h
fix warnings showing up with -Wall (#2692)
2025-10-24 11:43:35 -07:00
binary.cpp
Add Primitive::name and remove Primitive::print (#2365)
2025-07-14 14:06:35 -07:00
binary.h
Add Primitive::name and remove Primitive::print (#2365)
2025-07-14 14:06:35 -07:00
CMakeLists.txt
Centralize NAX condition (#2811)
2025-11-21 13:28:15 -08:00
compiled.cpp
Add Primitive::name and remove Primitive::print (#2365)
2025-07-14 14:06:35 -07:00
conv.cpp
faster depthwise 1D conv (#2567)
2025-09-08 11:37:23 -07:00
copy.cpp
[CUDA] Reduce use of managed memory (#2725)
2025-11-05 16:05:23 -08:00
custom_kernel.cpp
export custom kernel (#2756)
2025-11-13 11:29:50 -08:00
device.cpp
[CUDA] Tune ops per buffer based on device (#2761)
2025-11-16 06:29:49 -08:00
device.h
Centralize NAX condition (#2811)
2025-11-21 13:28:15 -08:00
distributed.cpp
Nccl reduce scatter, all gather (#2727)
2025-11-05 08:21:11 -08:00
eval.cpp
fix warnings showing up with -Wall (#2692)
2025-10-24 11:43:35 -07:00
event.cpp
[Metal] Release metal events (#2412)
2025-07-23 19:53:42 -07:00
fence.cpp
[CUDA] Reduce use of managed memory (#2725)
2025-11-05 16:05:23 -08:00
fft.cpp
[CUDA] Reduce use of managed memory (#2725)
2025-11-05 16:05:23 -08:00
hadamard.cpp
Move common gpu primitives to backend/gpu (#2145)
2025-05-05 13:45:29 -07:00
indexing.cpp
Add Masked Scatter (#2663)
2025-11-19 14:53:32 -08:00
jit_kernels.cpp
Add quantize/dequantize for mxfp8 and nvfp4 (#2688)
2025-10-28 16:23:12 -07:00
kernels.h
fix METAL quantization in JIT (#2553)
2025-08-28 18:26:25 -07:00
logsumexp.cpp
Add contiguous_copy_gpu util for copying array (#2379)
2025-07-18 06:44:25 -07:00
make_compiled_preamble.sh
Dispatch bf16 at run time when using the JIT (#1584)
2024-11-15 16:54:36 -08:00
matmul.cpp
Centralize NAX condition (#2811)
2025-11-21 13:28:15 -08:00
matmul.h
Collection of refactors (#2274)
2025-06-13 10:44:56 -07:00
metal.cpp
only build for macos 14 and up (#2731)
2025-11-04 09:44:15 -08:00
metal.h
Generalize gpu backend (#2138)
2025-04-30 09:08:17 -07:00
no_metal.cpp
Custom cuda kernel (#2517)
2025-08-20 17:20:22 -07:00
nojit_kernels.cpp
fix METAL quantization in JIT (#2553)
2025-08-28 18:26:25 -07:00
normalization.cpp
fix warnings showing up with -Wall (#2692)
2025-10-24 11:43:35 -07:00
primitives.cpp
fix warnings showing up with -Wall (#2692)
2025-10-24 11:43:35 -07:00
quantized.cpp
Centralize NAX condition (#2811)
2025-11-21 13:28:15 -08:00
reduce.cpp
Fix reduce sum/prod overflow (#2477)
2025-08-12 00:05:33 -07:00
reduce.h
Reductions update (#1351)
2024-11-04 22:25:16 -08:00
resident.cpp
Generalize gpu backend (#2138)
2025-04-30 09:08:17 -07:00
resident.h
Wired (#1510)
2024-10-25 09:35:33 -07:00
rope.cpp
no copy batch rope (#2595)
2025-09-15 14:23:48 -07:00
scaled_dot_product_attention.cpp
[CUDA] Support array mask in SDPA (#2822)
2025-11-26 11:08:58 +09:00
scan.cpp
Add Masked Scatter (#2663)
2025-11-19 14:53:32 -08:00
scan.h
Add Masked Scatter (#2663)
2025-11-19 14:53:32 -08:00
slicing.cpp
[CUDA] Implement DynamicSlice/DynamicSliceUpdate (#2533)
2025-08-26 07:31:39 +09:00
softmax.cpp
Add contiguous_copy_gpu util for copying array (#2379)
2025-07-18 06:44:25 -07:00
sort.cpp
Move common gpu primitives to backend/gpu (#2145)
2025-05-05 13:45:29 -07:00
ternary.cpp
Faster triu, tril, where with scalar (#2644)
2025-10-02 12:21:27 -07:00
ternary.h
Add Primitive::name and remove Primitive::print (#2365)
2025-07-14 14:06:35 -07:00
unary.cpp
Fp8 conversion (#2686)
2025-10-27 16:35:50 -07:00
unary.h
Add Primitive::name and remove Primitive::print (#2365)
2025-07-14 14:06:35 -07:00
utils.cpp
Move some dims utils to common (#2223)
2025-05-29 06:48:30 -07:00
utils.h
Add Primitive::name and remove Primitive::print (#2365)
2025-07-14 14:06:35 -07:00