Cheng 82faffa7bb Add more nccl stubs

2025-11-25 15:32:14 +09:00

..

3rdparty

jagrit's commit files

2023-11-29 10:52:08 -08:00

backend

Fix building when there is no GPU

2025-11-25 15:32:14 +09:00

distributed

Add more nccl stubs

2025-11-25 15:32:14 +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.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

[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