calculate_strides()

std::vector< uint64_t > MayaFlux::Kakshya::DataDimension::calculate_strides ( const std::vector< uint64_t > &  shape, MemoryLayout  layout  )

static

Calculate memory strides based on shape and layout.

Parameters

shape	Dimensional sizes
layout	Memory layout strategy

Returns

Vector of stride values for each dimension

Definition at line 271 of file NDData.cpp.

{
    if (shape.empty())
        return {};
 
    std::vector<uint64_t> strides(shape.size());
 
    if (layout == MemoryLayout::ROW_MAJOR) {
        auto reversed_shape = shape | std::views::reverse;
        std::exclusive_scan(
            reversed_shape.begin(),
            reversed_shape.end(),
            strides.rbegin(),
            uint64_t(1),
            std::multiplies<uint64_t> {});
    } else {
        std::exclusive_scan(
            shape.begin(),
            shape.end(),
            strides.begin(),
            uint64_t(1),
            std::multiplies<uint64_t> {});
    }
 
    return strides;
}

References MayaFlux::Kakshya::ROW_MAJOR.

Referenced by create_dimensions().

Here is the caller graph for this function: