cuda::experimental::places::make_partition_descriptor#

inline cute_partition_descriptor cuda::experimental::places::make_partition_descriptor(
dim4 true_dims,
const ::std::vector<dim_spec> &spec,
dim4 grid_dims
)#

Build a partition from a per-dimension specification.

Each entry of spec describes how the corresponding tensor dimension maps onto the grid (“blocked over axis 0”, …). Split dimensions are padded up to divisibility, which is what makes the resulting layout exact (see the file-level documentation). Every grid axis with extent > 1 must be bound by some entry; unbound axes would leave those places idle (replication is not supported) and are rejected at construction time.

Parameters:

  • true_dims – True tensor extents (dimension 0 fastest)

  • spec – One entry per tensor dimension (at most 4)

  • grid_dims – Extents of the grid of places