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

..

common

Fp64 on the CPU (#1843)

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

cpu

Double for lapack (#1904)

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

metal

Enabling fused attention for head dim 128 (#1899)

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

no_cpu

Bitwise Inverse (#1862)

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

no_metal

Bitwise Inverse (#1862)

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