Angelos Katharopoulos 9948eddf11 Fix nan and improve speed for qvm (#903)

2024-03-26 10:41:45 -07:00

..

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