Jagrit Digani 358e1fd6ab Fused GEMM (#1123) ...

* Basic gemm working

* Update addmm

* Clear out steel_gemm and steel_addmm kernels

* Fuse and clear out gather gemm

* Update objc releases

2024-05-15 10:30:41 -07:00

..

3rdparty

jagrit's commit files

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

backend

Fused GEMM (#1123)

2024-05-15 10:30:41 -07:00

io

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

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

types

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

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

allocator.cpp

Memory allocation (#292)

2024-01-02 11:59:19 -08:00

allocator.h

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

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

array.cpp

Fix leak for multi-output primitives which are never detached (#1059)

2024-05-01 07:31:45 -07:00

array.h

Fix leak for multi-output primitives which are never detached (#1059)

2024-05-01 07:31:45 -07:00

CMakeLists.txt

No CPU option for binary minimization (#1105)

2024-05-13 16:08:11 -07:00

compile_impl.h

Ios compile (#784)

2024-03-04 20:02:26 -08:00

compile.cpp

Add conjugate operator (#1100)

2024-05-10 07:22:20 -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

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

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

dtype.cpp

Remove unnecessary string copies (#891)

2024-03-28 13:14:59 -07:00

dtype.h

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

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

event.h

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

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

fast_primitives.h

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

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

fast.cpp

Option for precise softmax (#953)

2024-04-04 08:32:35 -07:00

fast.h

Adds mx.fast.layer_norm (#870)

2024-03-21 13:55:51 -07:00

fft.cpp

Don't use make_unique to create shared_ptr (#902)

2024-03-27 06:13:29 -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

use string (#976)

2024-04-09 11:22:00 -07:00

linalg.h

use string (#976)

2024-04-09 11:22:00 -07:00

mlx.h

Custom primitive + RoPE fat op (#676)

2024-02-14 14:04:25 -08:00

ops.cpp

Allow scatter type exception to be caught by checking in op (#1077)

2024-05-13 17:43:53 -07:00

ops.h

Conv3d (#993)

2024-05-11 06:15:02 -07:00

primitives.cpp

No CPU option for binary minimization (#1105)

2024-05-13 16:08:11 -07:00

primitives.h

Add conjugate operator (#1100)

2024-05-10 07:22:20 -07:00

random.cpp

Implementation of mlx.random.multivariate_normal (#502) (#877)

2024-04-09 13:50:12 -07:00

random.h

Implementation of mlx.random.multivariate_normal (#502) (#877)

2024-04-09 13:50:12 -07:00

scheduler.cpp

Add synchronize function (#1006)

2024-04-22 08:25:46 -07:00

scheduler.h

fix order device -> scheduler (#1039)

2024-04-26 13:46:41 -07:00

stream.h

Add synchronize function (#1006)

2024-04-22 08:25:46 -07:00

transforms_impl.h

Use uintptr_t instead of size_t to store funtion id (#916)

2024-03-28 06:37:59 -07:00

transforms.cpp

improve error messaging in eval (#1101)

2024-05-10 10:04:07 -07:00

transforms.h

Shared events for synchronization + async eval (#998)

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

utils.cpp

Shared events for synchronization + async eval (#998)

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

utils.h

Remove duplicate defines of StreamOrDevice and is_big_endian (#892)

2024-03-26 15:15:11 -07:00