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
997cfc7699e43e1a4650a98f18be8bd6a1885ac8
mlx/mlx/backend/cpu
History
Harsh Sutaria 618c87af8c
Some checks failed
Build and Test / check_lint (push) Has been cancelled
Details
Build and Test / linux_build_and_test (ubuntu-22.04) (push) Has been cancelled
Details
Build and Test / linux_build_and_test (ubuntu-22.04-arm) (push) Has been cancelled
Details
Build and Test / mac_build_and_test (14.0) (push) Has been cancelled
Details
Build and Test / mac_build_and_test (15.0) (push) Has been cancelled
Details
Build and Test / cuda_build_and_test (cuda-12.6) (push) Has been cancelled
Details
Build and Test / cuda_build_and_test (cuda-12.9) (push) Has been cancelled
Details
Build and Test / build_documentation (push) Has been cancelled
Details
Build and Test / Linux Fedora CPP Build (aarch64) (push) Has been cancelled
Details
Build and Test / Linux Fedora CPP Build (x86_64) (push) Has been cancelled
Details
Add float64 Eig and complex64 SVD/Eig support (Fixes #2708) (#2737) 
Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
Co-authored-by: Awni Hannun <awni@apple.com>
2025-11-22 06:51:36 -08:00
..
gemms
fix addmm cpu (#2699)
2025-10-27 11:33:32 -07:00
simd
Check isnan in maximum / minimum with CPU backend (#2652)
2025-11-03 08:51:14 -08:00
arange.h
redesign for faster cpu/gpu synch (#1869)
2025-03-06 19:23:38 -08:00
arg_reduce.cpp
Add remove_index utility (#2173)
2025-05-13 17:09:56 -07:00
available.cpp
Generalize gpu backend (#2138)
2025-04-30 09:08:17 -07:00
available.h
Generalize gpu backend (#2138)
2025-04-30 09:08:17 -07:00
binary_ops.h
Refactor common into cpu specific and truly common (#1817)
2025-02-03 15:58:02 -08:00
binary_two.h
reduce binary size (#1952)
2025-03-11 06:30:44 -07:00
binary.cpp
Make cpu binary_op easily accessible (#2733)
2025-11-05 01:08:41 -08:00
binary.h
Make cpu binary_op easily accessible (#2733)
2025-11-05 01:08:41 -08:00
cholesky.cpp
Rename the copy util in cpu/copy.h to copy_cpu (#2378)
2025-07-16 07:34:24 -07:00
CMakeLists.txt
non-symmetric eig and eigh (#2188)
2025-05-15 13:01:44 -07:00
compiled_preamble.h
Refactor common into cpu specific and truly common (#1817)
2025-02-03 15:58:02 -08:00
compiled.cpp
Separate cpu compilation cache by versions (#2548)
2025-08-27 11:25:15 +09:00
conv.cpp
fix warnings showing up with -Wall (#2692)
2025-10-24 11:43:35 -07:00
copy.cpp
Add contiguous_copy_cpu util for copying array (#2397)
2025-07-21 07:30:35 -07:00
copy.h
Add contiguous_copy_cpu util for copying array (#2397)
2025-07-21 07:30:35 -07:00
distributed.cpp
Nccl reduce scatter, all gather (#2727)
2025-11-05 08:21:11 -08:00
eig.cpp
Add float64 Eig and complex64 SVD/Eig support (Fixes #2708) (#2737)
2025-11-22 06:51:36 -08:00
eigh.cpp
Rename the copy util in cpu/copy.h to copy_cpu (#2378)
2025-07-16 07:34:24 -07:00
encoder.cpp
redesign for faster cpu/gpu synch (#1869)
2025-03-06 19:23:38 -08:00
encoder.h
Fix multistream GPU deadlock (#1969)
2025-03-20 07:19:47 -07:00
eval.cpp
Fix multistream GPU deadlock (#1969)
2025-03-20 07:19:47 -07:00
eval.h
redesign for faster cpu/gpu synch (#1869)
2025-03-06 19:23:38 -08:00
fft.cpp
fix malloc or wait deadlock (#1976)
2025-03-20 16:48:43 -07:00
gemm.h
redesign for faster cpu/gpu synch (#1869)
2025-03-06 19:23:38 -08:00
hadamard.cpp
Rename the copy util in cpu/copy.h to copy_cpu (#2378)
2025-07-16 07:34:24 -07:00
indexing.cpp
Add Masked Scatter (#2663)
2025-11-19 14:53:32 -08:00
inverse.cpp
Rename the copy util in cpu/copy.h to copy_cpu (#2378)
2025-07-16 07:34:24 -07:00
jit_compiler.cpp
Add missing <algorithm> header to jit_compiler.cpp (#2460)
2025-08-04 14:00:46 -07:00
jit_compiler.h
Refactor common into cpu specific and truly common (#1817)
2025-02-03 15:58:02 -08:00
lapack.h
Add float64 Eig and complex64 SVD/Eig support (Fixes #2708) (#2737)
2025-11-22 06:51:36 -08:00
logsumexp.cpp
Add contiguous_copy_cpu util for copying array (#2397)
2025-07-21 07:30:35 -07:00
luf.cpp
Rename the copy util in cpu/copy.h to copy_cpu (#2378)
2025-07-16 07:34:24 -07:00
make_compiled_preamble.ps1
Refactor common into cpu specific and truly common (#1817)
2025-02-03 15:58:02 -08:00
make_compiled_preamble.sh
fix cpu compile (#1897)
2025-02-24 14:10:30 -08:00
masked_mm.cpp
fix warnings showing up with -Wall (#2692)
2025-10-24 11:43:35 -07:00
matmul.cpp
Make cpu binary_op easily accessible (#2733)
2025-11-05 01:08:41 -08:00
primitives.cpp
[CUDA] Reduce use of managed memory (#2725)
2025-11-05 16:05:23 -08:00
qrf.cpp
Rename the copy util in cpu/copy.h to copy_cpu (#2378)
2025-07-16 07:34:24 -07:00
quantized.cpp
Fp8 conversion (#2686)
2025-10-27 16:35:50 -07:00
reduce.cpp
Fix reduce sum/prod overflow (#2477)
2025-08-12 00:05:33 -07:00
scan.cpp
Add contiguous_copy_cpu util for copying array (#2397)
2025-07-21 07:30:35 -07:00
select.cpp
reduce binary size (#1952)
2025-03-11 06:30:44 -07:00
slicing.h
Refactor common into cpu specific and truly common (#1817)
2025-02-03 15:58:02 -08:00
softmax.cpp
Add contiguous_copy_cpu util for copying array (#2397)
2025-07-21 07:30:35 -07:00
sort.cpp
fix warnings showing up with -Wall (#2692)
2025-10-24 11:43:35 -07:00
svd.cpp
Add float64 Eig and complex64 SVD/Eig support (Fixes #2708) (#2737)
2025-11-22 06:51:36 -08:00
ternary.h
reduce binary size (#1952)
2025-03-11 06:30:44 -07:00
threefry.cpp
Refactor common into cpu specific and truly common (#1817)
2025-02-03 15:58:02 -08:00
threefry.h
Refactor common into cpu specific and truly common (#1817)
2025-02-03 15:58:02 -08:00
unary_ops.h
Add quantize/dequantize for mxfp8 and nvfp4 (#2688)
2025-10-28 16:23:12 -07:00
unary.cpp
Fix MSVC build due to use of M_LN2 (#2058)
2025-04-10 07:41:41 -07:00
unary.h
Fp8 conversion (#2686)
2025-10-27 16:35:50 -07:00