.. |
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 |