cuda::experimental::places::static_cute_sub_shape

template<size_t rank, size_t num_leaves>
class static_cute_sub_shape

Compact sub-shape with a compile-time number of layout leaves.

This is the kernel-facing representation produced by a typed cute_partition. Unlike cute_sub_shape, it stores exactly the leaves used by the partition and has no runtime leaf count, so decoding can be unrolled.

Public Types

using coords_t = ::cuda::std::array<size_t, rank>

Public Functions

inline static_cute_sub_shape(

const ::cuda::std::array<layout_leaf, num_leaves> &local_leaves,

size_t offset,

dim4 padded_dims,

const ::cuda::std::array<size_t, rank> &lo,

const ::cuda::std::array<size_t, rank> &hi

)

inline bool contains(const coords_t &coords) const

inline size_t size() const

inline coords_t index_to_coords(size_t index) const