zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use SmallVector for shapes and strides

Browse Source
This commit is contained in:
Cheng
2025-08-01 21:38:10 +09:00
parent be9bc96da4
commit 68e9c60d22
30 changed files with 677 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
python/src/small_vector.h Normal file
View File

@@ -0,0 +1,17 @@
// Copyright © 2025 Apple Inc.
#pragma once
#include "mlx/small_vector.h"
#include <nanobind/stl/detail/nb_list.h>
NAMESPACE_BEGIN(NB_NAMESPACE)
NAMESPACE_BEGIN(detail)
template <typename Type, size_t Size, typename Alloc>
struct type_caster<mlx::core::SmallVector<Type, Size, Alloc>>
: list_caster<mlx::core::SmallVector<Type, Size, Alloc>, Type> {};
NAMESPACE_END(detail)
NAMESPACE_END(NB_NAMESPACE)