Angelos Katharopoulos 3f8b1668c4 Make reshape faster for row_contiguous cases (#829)

2024-03-13 16:22:03 -07:00

..

3rdparty

jagrit's commit files

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

backend

Make reshape faster for row_contiguous cases (#829)

2024-03-13 16:22:03 -07:00

io

Remove unused variables (#706)

2024-02-18 12:50:10 -08:00

types

Refactor CPU compile preamble (#708)

2024-02-19 06:12:53 -08: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

Compile stride bug (#812)

2024-03-11 06:31:31 -07:00

array.h

Compile stride bug (#812)

2024-03-11 06:31:31 -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

NumberOfElements for shapeless compile and vmap fixes (#802)

2024-03-13 10:34:14 -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

copyright + ack

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

dtype.h

Pickle + dtype fix for numpy conversion (#763)

2024-03-02 06:09:29 -08:00

fast_primitives.h

Fast Inference SDPA op (#735)

2024-03-04 21:06:11 -08:00

fast.cpp

route to fallback for bfloat (#794)

2024-03-06 15:39:12 -08:00

fast.h

Fast Inference SDPA op (#735)

2024-03-04 21:06:11 -08:00

fft.cpp

Spelling (#342)

2024-01-01 21:08:17 -08:00

fft.h

copyright + ack

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

graph_utils.cpp

Kernel generation (#614)

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

graph_utils.h

Kernel generation (#614)

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

io.h

Metadata support for safetensors (#639)

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

linalg.cpp

Add SVD primitive (#809)

2024-03-12 12:30:11 -07:00

linalg.h

Add SVD primitive (#809)

2024-03-12 12:30:11 -07:00

mlx.h

Custom primitive + RoPE fat op (#676)

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

ops.cpp

NumberOfElements for shapeless compile and vmap fixes (#802)

2024-03-13 10:34:14 -07:00

ops.h

NumberOfElements for shapeless compile and vmap fixes (#802)

2024-03-13 10:34:14 -07:00

primitives.cpp

NumberOfElements for shapeless compile and vmap fixes (#802)

2024-03-13 10:34:14 -07:00

primitives.h

NumberOfElements for shapeless compile and vmap fixes (#802)

2024-03-13 10:34:14 -07:00

random.cpp

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

2024-02-07 11:49:59 -08: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

NumberOfElements for shapeless compile and vmap fixes (#802)

2024-03-13 10:34:14 -07:00

transforms.h

Compile primitive (#571)

2024-02-05 06:51:22 -08:00

utils.cpp

Adds device context manager (#679)

2024-02-14 14:14:58 -08:00

utils.h

CPU compile (#691)

2024-02-17 06:54:32 -08:00