mlx/mlx/backend/metal/kernels
2024-03-12 11:42:07 -07:00
..
reduction Reduce update (#783) 2024-03-04 19:09:51 -08:00
steel Convolution update (#651) 2024-02-28 20:11:16 -08:00
arange.metal copyright + ack 2023-11-30 11:12:53 -08:00
arg_reduce.metal Fix threadgroup memory in arg reduce (#723) 2024-02-21 19:42:16 -08:00
atomic.h Fix eye for larger matrices (#463) 2024-01-16 00:51:24 -08:00
bf16_math.h copyright + ack 2023-11-30 11:12:53 -08:00
bf16.h copyright + ack 2023-11-30 11:12:53 -08:00
binary_two.metal Remainder negative numerator bug fixed (#641) 2024-02-09 16:49:14 -08:00
binary.h Remainder negative numerator bug fixed (#641) 2024-02-09 16:49:14 -08:00
binary.metal Fix logsumexp edge case (#740) 2024-02-25 08:39:55 -08:00
CMakeLists.txt Fast Inference SDPA op (#735) 2024-03-04 21:06:11 -08:00
compiled_preamble.h Implement the 'where' primitive for conditional selection (#664) 2024-02-22 15:10:48 -08:00
complex.h Remainder negative numerator bug fixed (#641) 2024-02-09 16:49:14 -08:00
conv.metal Convolution update (#651) 2024-02-28 20:11:16 -08:00
copy.metal copyright + ack 2023-11-30 11:12:53 -08:00
defines.h copyright + ack 2023-11-30 11:12:53 -08:00
erf.h copyright + ack 2023-11-30 11:12:53 -08:00
gather.metal Update gather and scatter to not use Argument Encoder (#683) 2024-02-14 13:42:13 -08:00
gemv.metal Fix oob reads in gemv kernel (#523) 2024-01-22 12:06:04 -08:00
indexing.h Update gather and scatter to not use Argument Encoder (#683) 2024-02-14 13:42:13 -08:00
quantized.metal Improved quantized matrix vector product (#786) 2024-03-05 17:32:19 -08:00
random.metal copyright + ack 2023-11-30 11:12:53 -08:00
rope.metal Custom primitive + RoPE fat op (#676) 2024-02-14 14:04:25 -08:00
scaled_dot_product_attention_params.h Fast Inference SDPA op (#735) 2024-03-04 21:06:11 -08:00
scaled_dot_product_attention.metal Compile stride bug (#812) 2024-03-11 06:31:31 -07:00
scan.metal copyright + ack 2023-11-30 11:12:53 -08:00
scatter.metal fix scatter (#821) 2024-03-12 11:42:07 -07:00
softmax.metal copyright + ack 2023-11-30 11:12:53 -08:00
sort.metal Spelling (#342) 2024-01-01 21:08:17 -08:00
ternary.h Implement the 'where' primitive for conditional selection (#664) 2024-02-22 15:10:48 -08:00
ternary.metal Fix logsumexp edge case (#740) 2024-02-25 08:39:55 -08:00
unary.h Fix logsumexp edge case (#740) 2024-02-25 08:39:55 -08:00
unary.metal Kernel generation (#614) 2024-02-07 13:15:59 -08:00
utils.h Implement the 'where' primitive for conditional selection (#664) 2024-02-22 15:10:48 -08:00