zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-07-19 15:41:13 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
a2ffea683a
mlx/mlx/backend/metal/kernels
History
Awni Hannun a2ffea683a
Fix eye for larger matrices (#463) 
* fix eye
* fix scatter for <32bit (non native atomic) types
* fix int overflow
2024-01-16 00:51:24 -08:00
..
gemm Spelling (#342) 2024-01-01 21:08:17 -08:00
arange.metal copyright + ack 2023-11-30 11:12:53 -08:00
arg_reduce.metal Spelling (#342) 2024-01-01 21:08:17 -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 Multi output primitives (#330) 2024-01-08 16:39:08 -08:00
binary.metal feat: add logicalAnd and logicalOR (#386) 2024-01-08 07:00:05 -08:00
CMakeLists.txt Fix eye for larger matrices (#463) 2024-01-16 00:51:24 -08:00
complex.h Spelling (#342) 2024-01-01 21:08:17 -08:00
conv_params.h copyright + ack 2023-11-30 11:12:53 -08:00
conv.metal copyright + ack 2023-11-30 11:12:53 -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
gemm.metal copyright + ack 2023-11-30 11:12:53 -08:00
gemv.metal Spelling (#342) 2024-01-01 21:08:17 -08:00
indexing.metal Fix eye for larger matrices (#463) 2024-01-16 00:51:24 -08:00
quantized.metal Allow arbitrary first dimension in quantization kernels. (#458) 2024-01-16 00:46:21 -08:00
random.metal copyright + ack 2023-11-30 11:12:53 -08:00
reduce.h Fix eye for larger matrices (#463) 2024-01-16 00:51:24 -08:00
reduce.metal Spelling (#342) 2024-01-01 21:08:17 -08:00
scan.metal copyright + ack 2023-11-30 11:12:53 -08: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
unary.metal Adds round op and primitive (#203) 2023-12-18 11:32:48 -08:00
utils.h copyright + ack 2023-11-30 11:12:53 -08:00