Cheng 828c5f1137 Use SmallVector for shapes and strides (#2454) ...

* Use SmallVector for shapes and strides

* Convert SmallVector to tuple

2025-08-05 09:41:03 +09:00

..

array.cpp

Use SmallVector for shapes and strides (#2454)

2025-08-05 09:41:03 +09:00

buffer.h

Double for lapack (#1904)

2025-02-25 11:39:36 -08:00

CMakeLists.txt

fix linux linking error (#2248)

2025-06-06 10:41:51 -07:00

constants.cpp

Array api (#1289)

2024-07-26 10:40:49 -07:00

convert.cpp

Compile float64 functions on CPU (#2311)

2025-06-24 10:18:52 -07:00

convert.h

Bump nanobind to 2.4 + fix (#1710)

2024-12-17 10:57:54 -08:00

device.cpp

start cuda circle config (#2256)

2025-06-10 21:19:47 -07:00

distributed.cpp

Use SmallVector for shapes and strides (#2454)

2025-08-05 09:41:03 +09:00

export.cpp

Use SmallVector for shapes and strides (#2454)

2025-08-05 09:41:03 +09:00

fast.cpp

Use SmallVector for shapes and strides (#2454)

2025-08-05 09:41:03 +09:00

fft.cpp

Use SmallVector for shapes and strides (#2454)

2025-08-05 09:41:03 +09:00

indexing.cpp

Dynamic slicing (#1741)

2025-01-07 14:02:16 -08:00

indexing.h

Remove "using namespace mlx::core" in python/src (#1689)

2024-12-11 15:45:39 -08:00

linalg.cpp

Use SmallVector for shapes and strides (#2454)

2025-08-05 09:41:03 +09:00

load.cpp

Use SmallVector for shapes and strides (#2454)

2025-08-05 09:41:03 +09:00

load.h

Remove "using namespace mlx::core" in python/src (#1689)

2024-12-11 15:45:39 -08:00

memory.cpp

move memory APIs into top level mlx.core (#1982)

2025-03-21 07:25:12 -07:00

metal.cpp

Use SmallVector for shapes and strides (#2454)

2025-08-05 09:41:03 +09:00

mlx_func.cpp

fix wraps compile (#2461)

2025-08-04 16:14:18 -07:00

mlx_func.h

fix wraps compile (#2461)

2025-08-04 16:14:18 -07:00

mlx.cpp

move memory APIs into top level mlx.core (#1982)

2025-03-21 07:25:12 -07:00

ops.cpp

Use SmallVector for shapes and strides (#2454)

2025-08-05 09:41:03 +09:00

random.cpp

Use SmallVector for shapes and strides (#2454)

2025-08-05 09:41:03 +09:00

small_vector.h

Use SmallVector for shapes and strides (#2454)

2025-08-05 09:41:03 +09:00

stream.cpp

Remove "using namespace mlx::core" in python/src (#1689)

2024-12-11 15:45:39 -08:00

transforms.cpp

Use SmallVector for shapes and strides (#2454)

2025-08-05 09:41:03 +09:00

trees.cpp

Limit grad recursion depth by not recursing through non-grad inputs (#1764)

2025-01-14 14:33:18 -08:00

trees.h

Limit grad recursion depth by not recursing through non-grad inputs (#1764)

2025-01-14 14:33:18 -08:00

utils.cpp

iinfo and scalar overflow detection (#2009)

2025-03-27 19:54:56 -07:00

utils.h

Bump nanobind to 2.4 + fix (#1710)

2024-12-17 10:57:54 -08:00