calculate_dispatch_size()

std::array< uint32_t, 3 > MayaFlux::Buffers::ComputeProcessor::calculate_dispatch_size ( const std::shared_ptr< VKBuffer > &  buffer )

protectedvirtual

Calculate dispatch size from buffer.

Parameters

buffer

Buffer to process

Returns

{group_count_x, group_count_y, group_count_z}

Override for custom dispatch calculation logic. Default implementation uses m_config.dispatch settings.

Reimplemented in MayaFlux::Buffers::VertexFieldProcessor.

Definition at line 142 of file ComputeProcessor.cpp.

{
    using DispatchMode = ShaderDispatchConfig::DispatchMode;
 
    switch (m_dispatch_config.mode) {
    case DispatchMode::MANUAL:
        return { m_dispatch_config.group_count_x, m_dispatch_config.group_count_y, m_dispatch_config.group_count_z };
 
    case DispatchMode::ELEMENT_COUNT: {
        uint64_t element_count = 0;
        const auto& dimensions = buffer->get_dimensions();
 
        if (!dimensions.empty()) {
            element_count = dimensions[0].size;
        } else {
            element_count = buffer->get_size_bytes() / sizeof(float);
        }
 
        auto groups_x = static_cast<uint32_t>(
            (element_count + m_dispatch_config.workgroup_x - 1) / m_dispatch_config.workgroup_x);
        return { groups_x, 1, 1 };
    }
 
    case DispatchMode::BUFFER_SIZE: {
        uint64_t size_bytes = buffer->get_size_bytes();
        auto groups_x = static_cast<uint32_t>(
            (size_bytes + m_dispatch_config.workgroup_x - 1) / m_dispatch_config.workgroup_x);
        return { groups_x, 1, 1 };
    }
 
    case DispatchMode::CUSTOM:
        if (m_dispatch_config.custom_calculator) {
            return m_dispatch_config.custom_calculator(buffer);
        }
        return { 1, 1, 1 };
 
    default:
        return { 1, 1, 1 };
    }
}

References MayaFlux::Buffers::ShaderDispatchConfig::custom_calculator, MayaFlux::Buffers::ShaderDispatchConfig::group_count_x, MayaFlux::Buffers::ShaderDispatchConfig::group_count_y, MayaFlux::Buffers::ShaderDispatchConfig::group_count_z, m_dispatch_config, MayaFlux::Buffers::ShaderDispatchConfig::mode, and MayaFlux::Buffers::ShaderDispatchConfig::workgroup_x.

Referenced by execute_shader().

Here is the caller graph for this function: