Angelos Katharopoulos 2e267bd6a8 Fix run without distributed

2024-09-06 11:03:01 -07:00

..

3rdparty

…

backend

Fix slice data size (#1394)

2024-09-04 19:10:43 -07:00

distributed

Fix run without distributed

2024-09-06 11:03:01 -07:00

io

Start a sockets based distributed backend

2024-09-06 11:03:01 -07:00

types

feat: metal formatting and pre-commit bump (#1038)

2024-04-30 07:18:09 -07:00

allocator.cpp

…

allocator.h

feat: metal formatting and pre-commit bump (#1038)

2024-04-30 07:18:09 -07:00

array.cpp

Fix leak with multi-output primitives (#1274)

2024-07-23 06:34:18 -07:00

array.h

Fix slice data size (#1394)

2024-09-04 19:10:43 -07:00

CMakeLists.txt

Einsum (#1269)

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

compile_impl.h

…

compile.cpp

Fixes segfault when compiling checkpointed functions (#1235)

2024-06-26 16:14:45 -07:00

compile.h

…

device.cpp

…

device.h

minor fixes (#1194)

2024-06-12 22:06:49 -07:00

dtype.cpp

Add gemv masked to JIT plus some fixes (#1310)

2024-08-07 13:38:07 -07:00

dtype.h

Add gemv masked to JIT plus some fixes (#1310)

2024-08-07 13:38:07 -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

minor fixes (#1194)

2024-06-12 22:06:49 -07:00

fast_primitives.h

Add grid_sample example to metal_kernel docs (#1352)

2024-08-23 18:24:16 -07:00

fast.cpp

Add optional headers to `mx.fast.metal_kernel` (#1358)

2024-08-26 21:45:45 -07:00

fast.h

Add optional headers to `mx.fast.metal_kernel` (#1358)

2024-08-26 21:45:45 -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

…

io.h

Add compiler flags to disable safetensors and gguf (#1098)

2024-05-09 17:39:44 -07:00

linalg.cpp

Pinv (#875)

2024-08-27 23:06:12 -07:00

linalg.h

Pinv (#875)

2024-08-27 23:06:12 -07:00

mlx.h

Einsum (#1269)

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

ops.cpp

Fix slice data size (#1394)

2024-09-04 19:10:43 -07:00

ops.h

Add isfinite (#1318)

2024-08-13 14:49:28 -07:00

primitives.cpp

Add "edge" mode to mx.pad (#1309)

2024-08-06 11:23:10 -07:00

primitives.h

CPU mx.linalg.cholesky_inverse and mx.linalg.tri_inv (#1307)

2024-08-08 15:18:02 -07:00

random.cpp

Einsum (#1269)

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

random.h

Implement sampling from laplace distribution. (#1279)

2024-07-24 15:15:37 +02: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

Custom transforms (#1246)

2024-07-10 18:00:01 -07:00

transforms.cpp

Make eval just wait if all arrays are scheduled (#1368)

2024-08-27 17:01:22 -07:00

transforms.h

Custom transforms (#1246)

2024-07-10 18:00:01 -07:00

utils.cpp

Shared events for synchronization + async eval (#998)

2024-04-17 06:16:02 -07:00

utils.h

MPI ops in GPU stream for faster comms (#1356)

2024-08-26 15:12:50 -07:00