mirror of
https://github.com/ml-explore/mlx.git
synced 2025-12-16 01:49:05 +08:00
18 lines
413 B
C++
18 lines
413 B
C++
// 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)
|