Jagrit Digani e21143961c Add init python binding for tunable matmul

2024-12-03 13:09:34 -08:00

..

3rdparty

jagrit's commit files

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

backend

CPP base for tunable matmul

2024-12-03 13:09:34 -08:00

distributed

Data parallel helper (#1407)

2024-09-16 18:17:21 -07:00

internal/tuner

Add cpp op and primitive for tunable matmul

2024-12-03 13:09:34 -08:00

io

contiguous op / prim (#1612)

2024-11-21 19:51:49 -08:00

types

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

2024-04-30 07:18:09 -07: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

Added missing unordered_map includes (#1635)

2024-12-02 07:03:03 -08:00

array.h

Fix array is_available race cases (#1468)

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

CMakeLists.txt

Add init python binding for tunable matmul

2024-12-03 13:09:34 -08: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

Cpu fast quantize (#1578)

2024-11-08 20:10:39 -08:00

fast.cpp

Matrix Attention kernel (#1610)

2024-11-22 10:34:05 -08:00

fast.h

3 and 6 bit quantization (#1613)

2024-11-22 10:22:13 -08: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

Added missing unordered_map includes (#1635)

2024-12-02 07:03:03 -08:00

io.h

Added missing unordered_map includes (#1635)

2024-12-02 07:03:03 -08: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

3 and 6 bit quantization (#1613)

2024-11-22 10:22:13 -08:00

ops.h

contiguous op / prim (#1612)

2024-11-21 19:51:49 -08:00

primitives.cpp

contiguous op / prim (#1612)

2024-11-21 19:51:49 -08:00

primitives.h

contiguous op / prim (#1612)

2024-11-21 19:51:49 -08: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

Enables build for *-linux-musl (#1627)

2024-11-27 13:14:24 -08:00

utils.h

Bfs width limit (#1568)

2024-11-08 15:00:46 -08:00