mlx/mlx/backend/metal/kernels/reduction
Awni Hannun 98b6ce3460
Refactor reductions and fix scatter atomics for large sizes (#1300)
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
2024-08-22 16:03:31 -07:00
..
ops.h Refactor reductions and fix scatter atomics for large sizes (#1300) 2024-08-22 16:03:31 -07:00
reduce_all.h Refactor reductions and fix scatter atomics for large sizes (#1300) 2024-08-22 16:03:31 -07:00
reduce_col.h Refactor reductions and fix scatter atomics for large sizes (#1300) 2024-08-22 16:03:31 -07:00
reduce_init.h More jitting (#1132) 2024-05-23 16:23:44 -07:00
reduce_row.h Refactor reductions and fix scatter atomics for large sizes (#1300) 2024-08-22 16:03:31 -07:00