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
1d4eacb737847d25a3b1960e84c84179a0950255
mlx/mlx
History
Andrey Portnoy 3e05cea9f8
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
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
Force cudaGraphExec reinstantiation when clusters are used (#2813) 
Co-authored-by: Awni Hannun <awni@apple.com>
2025-11-22 12:43:49 -08:00
..
3rdparty
jagrit's commit files
2023-11-29 10:52:08 -08:00
backend
Force cudaGraphExec reinstantiation when clusters are used (#2813)
2025-11-22 12:43:49 -08:00
distributed
Compatibility with pip-installed openmpi (#2741)
2025-11-07 16:58:31 -08: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.cpp
Add stats and limit to common allocator and enable tests (#1988)
2025-03-21 12:28:36 -07:00
allocator.h
[CUDA] Reduce use of managed memory (#2725)
2025-11-05 16:05:23 -08:00
array.cpp
fix slice (#2758)
2025-11-13 11:30:02 -08:00
array.h
fix slice (#2758)
2025-11-13 11:30:02 -08:00
CMakeLists.txt
start cuda circle config (#2256)
2025-06-10 21:19:47 -07:00
compile_impl.h
Fix compile multi capture (#2678)
2025-11-03 06:33:43 -08:00
compile.cpp
Fix compile multi capture (#2678)
2025-11-03 06:33:43 -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 exporting with constants (#2769)
2025-11-14 12:52:08 -08:00
export.h
export custom kernel (#2756)
2025-11-13 11:29:50 -08:00
fast_primitives.h
[CUDA] cuDNN backward attention (#2762)
2025-11-19 08:13:50 +09:00
fast.cpp
more accurate rope fallback (#2792)
2025-11-19 06:07:21 -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
Add Masked Scatter (#2663)
2025-11-19 14:53:32 -08:00
ops.h
Add Masked Scatter (#2663)
2025-11-19 14:53:32 -08:00
primitives.cpp
Add Masked Scatter (#2663)
2025-11-19 14:53:32 -08:00
primitives.h
Add Masked Scatter (#2663)
2025-11-19 14:53:32 -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
Generalize gpu backend (#2138)
2025-04-30 09:08:17 -07: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