Angelos Katharopoulos 99eefd2ec0 Gather mm new kernel and small refactoring (#2040)

2025-04-14 16:37:36 -07:00

..

3rdparty

jagrit's commit files

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

backend

Gather mm new kernel and small refactoring (#2040)

2025-04-14 16:37:36 -07:00

distributed

Fix the test and add custom min/max reductions for uncommon MPI types (#2060)

2025-04-10 17:01:17 -07:00

io

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

types

Fp64 on the CPU (#1843)

2025-02-07 15:52:22 -08:00

allocator.cpp

Add stats and limit to common allocator and enable tests (#1988)

2025-03-21 12:28:36 -07:00

allocator.h

Add stats and limit to common allocator and enable tests (#1988)

2025-03-21 12:28:36 -07:00

array.cpp

reduce binary size (#1952)

2025-03-11 06:30:44 -07:00

array.h

reduce binary size (#1952)

2025-03-11 06:30:44 -07:00

CMakeLists.txt

Do not define MLX_VERSION globally (#1966)

2025-03-18 07:12:40 -07:00

compile_impl.h

Simplify removes no-ops from the tape (#1759)

2025-01-09 11:23:19 -08:00

compile.cpp

Limit compile buffers (#1887)

2025-02-19 20:28:13 -08:00

compile.h

fix function pointer (#1865)

2025-02-13 18:46:11 -08:00

device.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

device.h

minor fixes (#1194)

2024-06-12 22:06:49 -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 ellipsis (#1788)

2025-01-25 01:28:03 -08:00

einsum.h

Einsum (#1269)

2024-07-25 09:36:44 -07:00

event.h

Remove Event::Signal() (#2052)

2025-04-08 06:20:27 -07:00

export_impl.h

Export / import functions to / from a file (#1642)

2024-12-24 11:19:13 -08:00

export.cpp

LogCumSumExp (#2069)

2025-04-13 01:27:29 -07:00

export.h

Export / import functions to / from a file (#1642)

2024-12-24 11:19:13 -08:00

fast_primitives.h

Support fused masking in Attention (#1924)

2025-03-20 11:01:32 -07:00

fast.cpp

Add float mask to sdpa vector (#2068)

2025-04-11 17:29:40 -07:00

fast.h

Add new sdpa function overload (#2035)

2025-04-03 11:58:28 -07:00

fence.h

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

fft.cpp

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

fft.h

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

graph_utils.cpp

Optionally specify names for arrays when exporting (#1749)

2025-01-06 13:07:46 -08: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

Adds nuclear norm support (#1894)

2025-03-04 13:26:02 -08:00

linalg.h

Adds nuclear norm support (#1894)

2025-03-04 13:26:02 -08:00

memory.h

move memory APIs into top level mlx.core (#1982)

2025-03-21 07:25:12 -07:00

mlx.h

move memory APIs into top level mlx.core (#1982)

2025-03-21 07:25:12 -07:00

ops.cpp

Gather mm new kernel and small refactoring (#2040)

2025-04-14 16:37:36 -07:00

ops.h

Gather mm new kernel and small refactoring (#2040)

2025-04-14 16:37:36 -07:00

primitives.cpp

Gather mm new kernel and small refactoring (#2040)

2025-04-14 16:37:36 -07:00

primitives.h

Gather mm new kernel and small refactoring (#2040)

2025-04-14 16:37:36 -07:00

random.cpp

Adds nuclear norm support (#1894)

2025-03-04 13:26:02 -08:00

random.h

Use int64 stride everywhere (#1671)

2024-12-09 11:09:02 -08:00

scheduler.cpp

Do not join threads during process exit on Windows (#1738)

2025-03-25 06:33:08 -07:00

scheduler.h

wait_for_one should wait for task to finish (#2047)

2025-04-05 20:05:16 -07: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

no sdpa in grad (#2054)

2025-04-08 19:13:54 -07:00

transforms.cpp

no sdpa in grad (#2054)

2025-04-08 19:13:54 -07:00

transforms.h

Export / import functions to / from a file (#1642)

2024-12-24 11:19:13 -08:00

utils.cpp

iinfo and scalar overflow detection (#2009)

2025-03-27 19:54:56 -07:00

utils.h

iinfo and scalar overflow detection (#2009)

2025-03-27 19:54:56 -07:00

version.cpp

Do not define MLX_VERSION globally (#1966)

2025-03-18 07:12:40 -07:00

version.h

patch bump (#2043)

2025-04-03 13:15:18 -07:00