Angelos Katharopoulos 9eb7d7362f Fix Split::vmap (#1845)

2025-02-08 09:22:13 -08:00

..

3rdparty

jagrit's commit files

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

backend

Fp64 on the CPU (#1843)

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

distributed

Fp64 on the CPU (#1843)

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

io

Ring distributed backend (#1784)

2025-01-27 22:15:01 -08:00

types

Fp64 on the CPU (#1843)

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

allocator.cpp

Fix sibling leak (#1590)

2024-11-18 19:17:01 -08:00

allocator.h

Allow querying the allocator for the buffer size (#1404)

2024-09-11 21:02:16 -07:00

array.cpp

Fp64 on the CPU (#1843)

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

array.h

Fp64 on the CPU (#1843)

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

CMakeLists.txt

Refactor common into cpu specific and truly common (#1817)

2025-02-03 15:58:02 -08:00

compile_impl.h

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

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

compile.cpp

recompile when shapeless is different (#1776)

2025-01-20 21:07:10 -08:00

compile.h

allow compiling lambdas in C++ (#1650)

2024-12-06 13:13:21 -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

Fp64 on the CPU (#1843)

2025-02-07 15:52:22 -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

Fix array is_available race cases (#1468)

2024-10-07 19:13:50 -07:00

export_impl.h

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

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

export.cpp

Dynamic broadcasting for shapeless compile/export (#1722)

2025-01-09 11:04:24 -08:00

export.h

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

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

fast_primitives.h

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

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

fast.cpp

Allow different value dimensions in sdpa_vector (#1811)

2025-01-31 20:58:59 -08:00

fast.h

Allow offset to be an mx.array for mx.fast.rope (#1724)

2024-12-19 15:51:44 -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

catch stream errors earlier to avoid aborts (#1801)

2025-01-27 14:05:43 -08:00

linalg.h

Eigenvalues and eigenvectors (#1334)

2024-10-22 12:18:48 -07:00

mlx.h

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

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

ops.cpp

Fix a couple of slicing bugs (#1827)

2025-02-05 19:50:08 -08:00

ops.h

scatter axis + gather axis primitives (#1813)

2025-01-31 20:48:08 -08:00

primitives.cpp

Fix Split::vmap (#1845)

2025-02-08 09:22:13 -08:00

primitives.h

Fix a couple of slicing bugs (#1827)

2025-02-05 19:50:08 -08:00

random.cpp

Use int64 stride everywhere (#1671)

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

random.h

Use int64 stride everywhere (#1671)

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

scheduler.cpp

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

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

scheduler.h

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

2024-12-24 11:19:13 -08: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

fix shapeless compile on ubuntu24 (#1775)

2025-01-18 06:04:36 -08:00

transforms.cpp

fix shapeless compile on ubuntu24 (#1775)

2025-01-18 06:04:36 -08:00

transforms.h

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

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

utils.cpp

Fp64 on the CPU (#1843)

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

utils.h

Fp64 on the CPU (#1843)

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