Jagrit Digani 89d327075f Enabling fused attention for head dim 128 (#1899) ...

* Share KV smem

* Fix bfloat error

* Unroll O = S @ V loop

* Perf upgrade

* Remove commented out function

* Add -Wno-c++17-extensions flag to metal flags

* Add -Wno-c++17-extensions flag to metal extension flags

2025-02-26 10:02:06 -08:00

..

3rdparty

jagrit's commit files

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

backend

Enabling fused attention for head dim 128 (#1899)

2025-02-26 10:02:06 -08:00

distributed

Fix ring of 2 and allow scalars in API (#1906)

2025-02-25 17:03:01 -08:00

io

add f8_e4m3 loading (#1859)

2025-02-13 17:10:03 -08:00

types

Fp64 on the CPU (#1843)

2025-02-07 15:52:22 -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

Fixes output donation for IO ops on the GPU (#1857)

2025-02-12 10:52:30 -08:00

array.h

Fixes output donation for IO ops on the GPU (#1857)

2025-02-12 10:52:30 -08:00

CMakeLists.txt

fix version and expose command queue getter (#1892)

2025-02-20 15:25:15 -08:00

compile_impl.h

Simplify removes no-ops from the tape (#1759)

2025-01-09 11:23:19 -08:00

compile.cpp

Limit compile buffers (#1887)

2025-02-19 20:28:13 -08:00

compile.h

fix function pointer (#1865)

2025-02-13 18:46:11 -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

fix double type promotion (#1901)

2025-02-25 06:00:53 -08:00

dtype.h

Fp64 on the CPU (#1843)

2025-02-07 15:52:22 -08: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

Added mlx::core::version() returning std::string(MLX_VERSION) (#1819)

2025-02-19 20:30:19 -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

Enabling fused attention for head dim 128 (#1899)

2025-02-26 10:02:06 -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

Double for lapack (#1904)

2025-02-25 11:39:36 -08:00

linalg.h

CPU LU factorization and linear solvers (#1451)

2025-02-10 12:32:24 -08:00

mlx.h

Added mlx::core::version() returning std::string(MLX_VERSION) (#1819)

2025-02-19 20:30:19 -08:00

ops.cpp

Ensure linspace always contains start and stop (#1883)

2025-02-19 13:53:20 -08:00

ops.h

Bitwise Inverse (#1862)

2025-02-13 08:44:14 -08:00

primitives.cpp

Bitwise Inverse (#1862)

2025-02-13 08:44:14 -08:00

primitives.h

Bitwise Inverse (#1862)

2025-02-13 08:44:14 -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

Fp64 on the CPU (#1843)

2025-02-07 15:52:22 -08:00

utils.h

Allow dynamic ops per buffer based on dispatches and memory (#1864)

2025-02-13 19:18:22 -08:00

version.cpp

fix version and expose command queue getter (#1892)

2025-02-20 15:25:15 -08:00

version.h

fix version and expose command queue getter (#1892)

2025-02-20 15:25:15 -08:00