mlx/mlx/backend
Awni Hannun 0c5eea226b
Reduce specializations (#1607)
* start of reduce specializations

* fix all reduce

* fix many dims

* fix

* non-jit tests clear

* cleanup instantiations

* cpu merges

* change dim specializations

* optimize

* fix jit

* fix jit

* use higher precision for integer sum+prod

* fixes
2024-11-21 19:53:00 -08:00
..
accelerate Eigenvalues and eigenvectors (#1334) 2024-10-22 12:18:48 -07:00
common Reduce specializations (#1607) 2024-11-21 19:53:00 -08:00
metal Reduce specializations (#1607) 2024-11-21 19:53:00 -08:00
no_cpu contiguous op / prim (#1612) 2024-11-21 19:51:49 -08:00
no_metal contiguous op / prim (#1612) 2024-11-21 19:51:49 -08:00