zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-14 17:12:49 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
Files
667c0f3bb993976ba5be04e14d1776b450e40aab
mlx/mlx
History
Awni Hannun 667c0f3bb9
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
[Metal] No copy array init (#2875)
2025-12-05 13:36:45 -08:00
..
3rdparty
jagrit's commit files
2023-11-29 10:52:08 -08:00
backend
[Metal] No copy array init (#2875)
2025-12-05 13:36:45 -08:00
distributed
Merge build-cuda and build-linux actions (#2783)
2025-11-25 20:06:42 +09:00
io
[CUDA] Reduce use of managed memory (#2725)
2025-11-05 16:05:23 -08:00
types
fix pinv (#2110)
2025-04-23 13:08:28 -07:00
allocator.h
[Metal] No copy array init (#2875)
2025-12-05 13:36:45 -08:00
array.cpp
[Metal] No copy array init (#2875)
2025-12-05 13:36:45 -08:00
array.h
[Metal] No copy array init (#2875)
2025-12-05 13:36:45 -08:00
CMakeLists.txt
Make allocator::malloc throw on allocation failure (#2874)
2025-12-05 17:44:38 +09:00
compile_impl.h
Fix compile multi capture (#2678)
2025-11-03 06:33:43 -08:00
compile.cpp
fix compile copying (#2871)
2025-12-04 12:32:56 -08:00
compile.h
fix function pointer (#1865)
2025-02-13 18:46:11 -08:00
device.cpp
Generalize gpu backend (#2138)
2025-04-30 09:08:17 -07:00
device.h
Generalize gpu backend (#2138)
2025-04-30 09:08:17 -07:00
dtype_utils.cpp
MLX_SWITCH macros to templates (#2320)
2025-07-01 01:33:44 -07:00
dtype_utils.h
Faster complex matmul (#2571)
2025-10-02 23:33:15 -07:00
dtype.cpp
fix double type promotion (#1901)
2025-02-25 06:00:53 -08:00
dtype.h
Fp64 on the CPU (#1843)
2025-02-07 15:52:22 -08:00
einsum.cpp
Einsum error msg improvement (#2690)
2025-10-27 06:31:47 -07:00
einsum.h
Einsum (#1269)
2024-07-25 09:36:44 -07:00
event.h
fix: linux-{fedora}x86_64-build (#2707)
2025-10-27 16:36:08 -07:00
export_impl.h
Fix exporting with constants (#2769)
2025-11-14 12:52:08 -08:00
export.cpp
Fix export scatters (#2852)
2025-12-02 11:24:40 -08:00
export.h
export custom kernel (#2756)
2025-11-13 11:29:50 -08:00
fast_primitives.h
[CUDA] Support array mask in SDPA (#2822)
2025-11-26 11:08:58 +09:00
fast.cpp
Layer norm throws on dimension mismatch (#2870)
2025-12-04 11:21:05 -08:00
fast.h
Use std::optional for mask_arr arg (#2763)
2025-11-17 10:43:33 +09:00
fence.h
[CUDA] Reduce use of managed memory (#2725)
2025-11-05 16:05:23 -08:00
fft.cpp
add fftshift and ifftshift fft helpers (#2135)
2025-04-29 22:13:45 -07:00
fft.h
add fftshift and ifftshift fft helpers (#2135)
2025-04-29 22:13:45 -07:00
graph_utils.cpp
Add Primitive::name and remove Primitive::print (#2365)
2025-07-14 14:06:35 -07:00
graph_utils.h
Optionally specify names for arrays when exporting (#1749)
2025-01-06 13:07:46 -08:00
io.h
Added missing unordered_map includes (#1635)
2024-12-02 07:03:03 -08:00
linalg.cpp
Add float64 Eig and complex64 SVD/Eig support (Fixes #2708) (#2737)
2025-11-22 06:51:36 -08:00
linalg.h
non-symmetric eig and eigh (#2188)
2025-05-15 13:01:44 -07:00
memory.h
move memory APIs into top level mlx.core (#1982)
2025-03-21 07:25:12 -07:00
mlx.h
Run CPP tests for CUDA build in CI (#2544)
2025-08-27 08:06:46 +09:00
ops.cpp
Support more Numpy interfaces for masked_scatter (#2832)
2025-12-01 17:51:02 -08:00
ops.h
Add Masked Scatter (#2663)
2025-11-19 14:53:32 -08:00
primitives.cpp
Reduce JVP (#2854)
2025-12-02 16:17:47 -08:00
primitives.h
Reduce JVP (#2854)
2025-12-02 16:17:47 -08:00
random.cpp
lower memory uniform sampling (#2361)
2025-07-15 14:22:07 -07:00
random.h
Add random normal distribution for complex numbers (#2182)
2025-05-13 22:43:45 -07:00
scheduler.cpp
Generalize gpu backend (#2138)
2025-04-30 09:08:17 -07:00
scheduler.h
[CUDA] Exit on crash and more helpful errors (#2830)
2025-11-24 19:46:03 -08:00
small_vector.h
Clean up code handling both std::vector and SmallVector (#2493)
2025-08-16 09:01:10 +09:00
stream.h
Export / import functions to / from a file (#1642)
2024-12-24 11:19:13 -08:00
threadpool.h
Ring distributed backend (#1784)
2025-01-27 22:15:01 -08:00
transforms_impl.h
Remove static initializers (#2059)
2025-04-24 06:14:49 -07:00
transforms.cpp
[CUDA] Reduce use of managed memory (#2725)
2025-11-05 16:05:23 -08:00
transforms.h
Export / import functions to / from a file (#1642)
2024-12-24 11:19:13 -08:00
utils.cpp
fix warnings showing up with -Wall (#2692)
2025-10-24 11:43:35 -07:00
utils.h
Nccl timeout (#2673)
2025-10-14 12:29:54 -07:00
version.cpp
Fix a few ccache cache miss (#2573)
2025-09-09 07:41:05 +09:00
version.h
patch bump for future version (#2804)
2025-11-20 09:26:20 -08:00