| .. |
|
3rdparty
|
|
|
|
backend
|
Remove the hack around SmallVector in cpu compile (#2494)
|
2025-08-16 08:17:24 +09:00 |
|
distributed
|
Add Primitive::name and remove Primitive::print (#2365)
|
2025-07-14 14:06:35 -07:00 |
|
io
|
|
|
|
types
|
|
|
|
allocator.cpp
|
|
|
|
allocator.h
|
|
|
|
array.cpp
|
|
|
|
array.h
|
Use SmallVector for shapes and strides (#2454)
|
2025-08-05 09:41:03 +09:00 |
|
CMakeLists.txt
|
start cuda circle config (#2256)
|
2025-06-10 21:19:47 -07:00 |
|
compile_impl.h
|
|
|
|
compile.cpp
|
Add Primitive::name and remove Primitive::print (#2365)
|
2025-07-14 14:06:35 -07:00 |
|
compile.h
|
|
|
|
device.cpp
|
|
|
|
device.h
|
|
|
|
dtype_utils.cpp
|
MLX_SWITCH macros to templates (#2320)
|
2025-07-01 01:33:44 -07:00 |
|
dtype_utils.h
|
MLX_SWITCH macros to templates (#2320)
|
2025-07-01 01:33:44 -07:00 |
|
dtype.cpp
|
|
|
|
dtype.h
|
|
|
|
einsum.cpp
|
|
|
|
einsum.h
|
|
|
|
event.h
|
|
|
|
export_impl.h
|
|
|
|
export.cpp
|
Use SmallVector for shapes and strides (#2454)
|
2025-08-05 09:41:03 +09:00 |
|
export.h
|
|
|
|
fast_primitives.h
|
Add Primitive::name and remove Primitive::print (#2365)
|
2025-07-14 14:06:35 -07:00 |
|
fast.cpp
|
full row mask in sdpa consistently gives nan (#2406)
|
2025-07-23 16:37:03 -07:00 |
|
fast.h
|
|
|
|
fence.h
|
|
|
|
fft.cpp
|
|
|
|
fft.h
|
|
|
|
graph_utils.cpp
|
Add Primitive::name and remove Primitive::print (#2365)
|
2025-07-14 14:06:35 -07:00 |
|
graph_utils.h
|
|
|
|
io.h
|
|
|
|
linalg.cpp
|
Add Primitive::name and remove Primitive::print (#2365)
|
2025-07-14 14:06:35 -07:00 |
|
linalg.h
|
|
|
|
memory.h
|
|
|
|
mlx.h
|
start cuda circle config (#2256)
|
2025-06-10 21:19:47 -07:00 |
|
ops.cpp
|
Fix logsumexp/softmax not fused for some cases (#2474)
|
2025-08-08 14:07:17 -07:00 |
|
ops.h
|
MoE backward improvements (#2335)
|
2025-07-07 17:59:53 -07:00 |
|
primitives.cpp
|
Use SmallVector for shapes and strides (#2454)
|
2025-08-05 09:41:03 +09:00 |
|
primitives.h
|
Use SmallVector for shapes and strides (#2454)
|
2025-08-05 09:41:03 +09:00 |
|
random.cpp
|
lower memory uniform sampling (#2361)
|
2025-07-15 14:22:07 -07:00 |
|
random.h
|
|
|
|
scheduler.cpp
|
|
|
|
scheduler.h
|
|
|
|
small_vector.h
|
Remove the hack around SmallVector in cpu compile (#2494)
|
2025-08-16 08:17:24 +09:00 |
|
stream.h
|
|
|
|
threadpool.h
|
|
|
|
transforms_impl.h
|
|
|
|
transforms.cpp
|
[Metal] Release metal events (#2412)
|
2025-07-23 19:53:42 -07:00 |
|
transforms.h
|
|
|
|
utils.cpp
|
Use SmallVector for shapes and strides (#2454)
|
2025-08-05 09:41:03 +09:00 |
|
utils.h
|
Use SmallVector for shapes and strides (#2454)
|
2025-08-05 09:41:03 +09:00 |
|
version.cpp
|
|
|
|
version.h
|
version (#2470)
|
2025-08-07 00:36:04 -07:00 |
