mlx/mlx
Awni Hannun af1b725fda
Fix a couple of slicing bugs (#1827)
* fix a few bugs

* fix conv grad

* speedup test

* comment
2025-02-05 19:50:08 -08:00
..
3rdparty jagrit's commit files 2023-11-29 10:52:08 -08:00
backend Fix a couple of slicing bugs (#1827) 2025-02-05 19:50:08 -08:00
distributed Refactor common into cpu specific and truly common (#1817) 2025-02-03 15:58:02 -08:00
io Ring distributed backend (#1784) 2025-01-27 22:15:01 -08:00
types Start to cleanup/unify accelerate and common back-ends (Part 1/N) (#1777) 2025-01-29 14:34:49 -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 Dynamic broadcasting for shapeless compile/export (#1722) 2025-01-09 11:04:24 -08:00
array.h Dynamic slicing (#1741) 2025-01-07 14:02:16 -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 dtype is copy assignable (#1436) 2024-09-25 12:07:13 -07:00
dtype.h dtype is copy assignable (#1436) 2024-09-25 12:07:13 -07: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 a couple of slicing bugs (#1827) 2025-02-05 19:50:08 -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 Dynamic broadcasting for shapeless compile/export (#1722) 2025-01-09 11:04:24 -08:00
utils.h Faster synchronization Fence primitive (#1773) 2025-01-17 18:42:19 -08:00