calculate_dispatch_size()

std::array< uint32_t, 3 > MayaFlux::Yantra::GpuDispatchCore::calculate_dispatch_size ( size_t  total_elements, const DataStructureInfo &  structure_info  ) const

protectedvirtual

Calculate workgroup dispatch counts from structure dimensions.

Reads SPATIAL_X/Y/Z roles for 2D/3D shaders; falls back to 1D element-count dispatch when no spatial dimensions exist.

Parameters

total_elements	Flat element count for the 1D fallback.
structure_info	Dimension metadata.

Reimplemented in MayaFlux::Yantra::TextureExecutionContext.

Definition at line 193 of file GpuDispatchCore.cpp.

{
    uint64_t sz_x = 0, sz_y = 0, sz_z = 0;
    for (const auto& dim : structure_info.dimensions) {
        switch (dim.role) {
        case Kakshya::DataDimension::Role::SPATIAL_X:
            sz_x = dim.size;
            break;
        case Kakshya::DataDimension::Role::SPATIAL_Y:
            sz_y = dim.size;
            break;
        case Kakshya::DataDimension::Role::SPATIAL_Z:
            sz_z = dim.size;
            break;
        default:
            break;
        }
    }
 
    const auto& ws = m_gpu_config.workgroup_size;
    if (sz_x > 0) {
        return {
            static_cast<uint32_t>((sz_x + ws[0] - 1) / ws[0]),
            sz_y > 0 ? static_cast<uint32_t>((sz_y + ws[1] - 1) / ws[1]) : 1U,
            sz_z > 0 ? static_cast<uint32_t>((sz_z + ws[2] - 1) / ws[2]) : 1U,
        };
    }
 
    return { static_cast<uint32_t>((total_elements + ws[0] - 1) / ws[0]), 1U, 1U };
}

References MayaFlux::Yantra::DataStructureInfo::dimensions, m_gpu_config, MayaFlux::Kakshya::DataDimension::SPATIAL_X, MayaFlux::Kakshya::DataDimension::SPATIAL_Y, MayaFlux::Kakshya::DataDimension::SPATIAL_Z, and MayaFlux::Yantra::GpuShaderConfig::workgroup_size.

Referenced by dispatch_core(), and dispatch_core_chained().

Here is the caller graph for this function: