| .. |
|
kernels
|
Fix nan and improve speed for qvm (#903)
|
2024-03-26 10:41:45 -07:00 |
|
mps
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
allocator.cpp
|
Some fixes in cache / thread safety (#777)
|
2024-03-05 13:30:50 -08:00 |
|
allocator.h
|
Some fixes in cache / thread safety (#777)
|
2024-03-05 13:30:50 -08:00 |
|
CMakeLists.txt
|
Adds mx.fast.layer_norm (#870)
|
2024-03-21 13:55:51 -07:00 |
|
compiled_preamble.h
|
Kernel generation (#614)
|
2024-02-07 13:15:59 -08:00 |
|
compiled.cpp
|
Reshape improvement (#818)
|
2024-03-12 17:54:31 -07:00 |
|
conv.cpp
|
No copy gems (#801)
|
2024-03-12 13:13:41 -07:00 |
|
copy.cpp
|
Fix copy donation and add partial rope (#881)
|
2024-03-22 17:28:26 -07:00 |
|
copy.h
|
Add a SliceUpdate op and primitive (#850)
|
2024-03-20 10:39:25 -07:00 |
|
device.cpp
|
Some C++ code are not needed (#841)
|
2024-03-18 17:04:10 -07:00 |
|
device.h
|
Kernel generation (#614)
|
2024-02-07 13:15:59 -08:00 |
|
fft.cpp
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
indexing.cpp
|
Some C++ code are not needed (#841)
|
2024-03-18 17:04:10 -07:00 |
|
make_compiled_preamble.sh
|
quote file name (#670)
|
2024-02-11 10:33:30 -08:00 |
|
matmul.cpp
|
add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427)
|
2024-03-25 12:32:59 -07:00 |
|
matmul.h
|
No copy gems (#801)
|
2024-03-12 13:13:41 -07:00 |
|
metal.cpp
|
Fix some issues using MLX in C++ (#739)
|
2024-02-24 22:20:57 -08:00 |
|
metal.h
|
bindings for memory info (#761)
|
2024-03-01 19:51:58 -08:00 |
|
normalization.cpp
|
Adds mx.fast.layer_norm (#870)
|
2024-03-21 13:55:51 -07:00 |
|
primitives.cpp
|
add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427)
|
2024-03-25 12:32:59 -07:00 |
|
quantized.cpp
|
Fix nan and improve speed for qvm (#903)
|
2024-03-26 10:41:45 -07:00 |
|
reduce.cpp
|
Strided reduce specialization for small reductions (#826)
|
2024-03-14 09:16:53 -07:00 |
|
rope.cpp
|
Fix copy donation and add partial rope (#881)
|
2024-03-22 17:28:26 -07:00 |
|
scaled_dot_product_attention.cpp
|
add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427)
|
2024-03-25 12:32:59 -07:00 |
|
scan.cpp
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
softmax.cpp
|
add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427)
|
2024-03-25 12:32:59 -07:00 |
|
sort.cpp
|
Spelling (#342)
|
2024-01-01 21:08:17 -08:00 |
|
utils.h
|
Add a SliceUpdate op and primitive (#850)
|
2024-03-20 10:39:25 -07:00 |
