Angelos Katharopoulos aca7584635 Fix OOB read in qmv when non-divisible by blocksize (#917)

2024-03-27 22:18:35 -07:00

..

3rdparty

jagrit's commit files

2023-11-29 10:52:08 -08:00

backend

Fix OOB read in qmv when non-divisible by blocksize (#917)

2024-03-27 22:18:35 -07:00

io

Don't use make_unique to create shared_ptr (#902)

2024-03-27 06:13:29 -07:00

types

Do not use static constexpr in header (#863)

2024-03-20 21:28:05 -07:00

allocator.cpp

Memory allocation (#292)

2024-01-02 11:59:19 -08:00

allocator.h

Memory allocation (#292)

2024-01-02 11:59:19 -08:00

array.cpp

Reduce a little overhead (#871)

2024-03-22 17:29:36 -07:00

array.h

Fix race in multi-stream eval (#911)

2024-03-26 16:36:36 -07:00

CMakeLists.txt

Custom primitive + RoPE fat op (#676)

2024-02-14 14:04:25 -08:00

compile_impl.h

Ios compile (#784)

2024-03-04 20:02:26 -08:00

compile.cpp

Reduce implicit copies in make_array (#874)

2024-03-22 06:29:16 -07:00

compile.h

Shapeless compilation for some graphs (#687)

2024-02-19 21:43:54 -08:00

device.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

device.h

copyright + ack

2023-11-30 11:12:53 -08:00

dtype.cpp

Remove duplicate defines of StreamOrDevice and is_big_endian (#892)

2024-03-26 15:15:11 -07:00

dtype.h

add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427)

2024-03-25 12:32:59 -07:00

fast_primitives.h

Implement vjps for some primitives in the fast namespace (#883)

2024-03-26 16:35:34 -07:00

fast.cpp

Implement vjps for some primitives in the fast namespace (#883)

2024-03-26 16:35:34 -07:00

fast.h

Adds mx.fast.layer_norm (#870)

2024-03-21 13:55:51 -07:00

fft.cpp

Don't use make_unique to create shared_ptr (#902)

2024-03-27 06:13:29 -07:00

fft.h

Remove duplicate defines of StreamOrDevice and is_big_endian (#892)

2024-03-26 15:15:11 -07:00

graph_utils.cpp

Kernel generation (#614)

2024-02-07 13:15:59 -08:00

graph_utils.h

Add missing && when forwarding args (#894)

2024-03-25 14:55:54 -07:00

io.h

Metadata support for safetensors (#639)

2024-02-08 19:33:15 -08:00

linalg.cpp

add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427)

2024-03-25 12:32:59 -07:00

linalg.h

Add matrix inversion primitive (#822)

2024-03-15 06:34:36 -07:00

mlx.h

Custom primitive + RoPE fat op (#676)

2024-02-14 14:04:25 -08:00

ops.cpp

Implement negative padding in conv with slicing (#907)

2024-03-26 14:59:19 -07:00

ops.h

Reduce a little overhead (#871)

2024-03-22 17:29:36 -07:00

primitives.cpp

Don't use make_unique to create shared_ptr (#902)

2024-03-27 06:13:29 -07:00

primitives.h

Add vmap for SVD and inverse (#849)

2024-03-21 13:18:27 -07:00

random.cpp

Fixing random.normal for half-precision dtype #642 (#904)

2024-03-26 09:58:27 -07:00

random.h

Add loc and scale to random.normal (#638)

2024-02-07 11:49:59 -08:00

scheduler.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

scheduler.h

move all ObjC (via metal-cpp) interaction until post static initializers (#370)

2024-01-04 16:12:00 -08:00

stream.h

copyright + ack

2023-11-30 11:12:53 -08:00

transforms_impl.h

Ios compile (#784)

2024-03-04 20:02:26 -08:00

transforms.cpp

Don't use make_unique to create shared_ptr (#902)

2024-03-27 06:13:29 -07:00

transforms.h

Add missing && when forwarding args (#894)

2024-03-25 14:55:54 -07:00

utils.cpp

Add a SliceUpdate op and primitive (#850)

2024-03-20 10:39:25 -07:00

utils.h

Remove duplicate defines of StreamOrDevice and is_big_endian (#892)

2024-03-26 15:15:11 -07:00