mlx/mlx
Alex Barron a4c47b0276
OOB QMV fix (#1579)
* fix oob access in qmv

* skip more

* fix small case
2024-11-08 17:59:45 -08:00
..
3rdparty jagrit's commit files 2023-11-29 10:52:08 -08:00
backend OOB QMV fix (#1579) 2024-11-08 17:59:45 -08:00
distributed Data parallel helper (#1407) 2024-09-16 18:17:21 -07:00
io A few updates for CPU (#1482) 2024-10-14 12:45:49 -07:00
types feat: metal formatting and pre-commit bump (#1038) 2024-04-30 07:18:09 -07:00
allocator.cpp Allow querying the allocator for the buffer size (#1404) 2024-09-11 21:02:16 -07:00
allocator.h Allow querying the allocator for the buffer size (#1404) 2024-09-11 21:02:16 -07:00
array.cpp fix multi output leak (#1548) 2024-10-31 09:32:01 -07:00
array.h Fix array is_available race cases (#1468) 2024-10-07 19:13:50 -07:00
CMakeLists.txt Chore: add pre-commit hook for cmake (#1362) 2024-09-16 12:53:01 -07:00
compile_impl.h Fix cpu segfault (#1488) 2024-10-14 16:17:03 -07:00
compile.cpp Real and Imag (#1490) 2024-10-15 16:23:15 -07:00
compile.h Shapeless compilation for some graphs (#687) 2024-02-19 21:43:54 -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 (#1269) 2024-07-25 09:36:44 -07: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
fast_primitives.h Some overhead reductions in mx.fast.metal_kernel (#1437) 2024-09-25 17:25:21 -07:00
fast.cpp fix per-example mask + docs in sdpa (#1574) 2024-11-08 11:51:15 -08:00
fast.h Simplifications for MLX C (#1396) 2024-09-06 19:16:50 -07:00
fft.cpp Feature complete Metal FFT (#1102) 2024-06-06 12:57:25 -07:00
fft.h Remove duplicate defines of StreamOrDevice and is_big_endian (#892) 2024-03-26 15:15:11 -07:00
graph_utils.cpp Remove unnecessary string copies (#891) 2024-03-28 13:14:59 -07:00
graph_utils.h Add missing && when forwarding args (#894) 2024-03-25 14:55:54 -07:00
io.h Add compiler flags to disable safetensors and gguf (#1098) 2024-05-09 17:39:44 -07:00
linalg.cpp Eigenvalues and eigenvectors (#1334) 2024-10-22 12:18:48 -07:00
linalg.h Eigenvalues and eigenvectors (#1334) 2024-10-22 12:18:48 -07:00
mlx.h Einsum (#1269) 2024-07-25 09:36:44 -07:00
ops.cpp add groups in conv2d (#1569) 2024-11-07 13:57:53 -08:00
ops.h Batched Quantized Matmul + Fast Small QMV (#1503) 2024-10-21 16:23:17 -07:00
primitives.cpp Fix gather vmap (#1563) 2024-11-05 11:29:20 -08:00
primitives.h Eigenvalues and eigenvectors (#1334) 2024-10-22 12:18:48 -07:00
random.cpp Faster bits and bernoulli (#1535) 2024-10-28 11:11:00 -07:00
random.h Add mx.random.permutation (#1471) 2024-10-08 19:42:19 -07:00
scheduler.cpp Add synchronize function (#1006) 2024-04-22 08:25:46 -07:00
scheduler.h minor fixes (#1194) 2024-06-12 22:06:49 -07:00
stream.h Add synchronize function (#1006) 2024-04-22 08:25:46 -07:00
transforms_impl.h Fix cpu segfault (#1488) 2024-10-14 16:17:03 -07:00
transforms.cpp Bfs width limit (#1568) 2024-11-08 15:00:46 -08:00
transforms.h Custom transforms (#1246) 2024-07-10 18:00:01 -07:00
utils.cpp Bfs width limit (#1568) 2024-11-08 15:00:46 -08:00
utils.h Bfs width limit (#1568) 2024-11-08 15:00:46 -08:00