dispatch()

void MayaFlux::Yantra::GpuResourceManager::dispatch	(	const std::string &	key,
		const std::array< uint32_t, 3 > &	groups,
		const std::vector< GpuBufferBinding > &	bindings,
		const uint8_t *	push_constant_data,
		size_t	push_constant_size
	)

Definition at line 474 of file GpuResourceManager.cpp.

{
    auto& unit = unit_for(key);
    auto& foundry = Portal::Graphics::get_shader_foundry();
    auto& compute_press = Portal::Graphics::get_compute_press();
 
    auto cmd_id = foundry.begin_commands(
        Portal::Graphics::ShaderFoundry::CommandBufferType::COMPUTE);
 
    compute_press.bind_all(
        cmd_id, unit.pipeline_id, unit.descriptor_set_ids,
        push_constant_data, push_constant_size);
 
    compute_press.dispatch(cmd_id, groups[0], groups[1], groups[2]);
 
    for (const auto& b : bindings) {
        const auto et = b.element_type;
        const bool is_image = et == GpuBufferBinding::ElementType::IMAGE_STORAGE
            || et == GpuBufferBinding::ElementType::IMAGE_SAMPLED;
        const bool is_output = b.direction == GpuBufferBinding::Direction::OUTPUT
            || b.direction == GpuBufferBinding::Direction::INPUT_OUTPUT;
 
        const bool is_shared = m_shared->slots.contains({ b.set, static_cast<size_t>(b.binding) });
        if (is_output && !is_image && !is_shared
            && static_cast<size_t>(b.binding) < unit.impl->buffers.size()) {
            foundry.buffer_barrier(
                cmd_id,
                unit.impl->buffers[b.binding].buffer,
                vk::AccessFlagBits::eShaderWrite,
                vk::AccessFlagBits::eHostRead,
                vk::PipelineStageFlagBits::eComputeShader,
                vk::PipelineStageFlagBits::eHost);
        }
    }
 
    foundry.submit_and_wait(cmd_id);
}

References b, MayaFlux::Portal::Graphics::ShaderFoundry::COMPUTE, MayaFlux::Portal::Graphics::get_compute_press(), MayaFlux::Portal::Graphics::get_shader_foundry(), MayaFlux::Portal::Graphics::GpuBufferBinding::IMAGE_SAMPLED, MayaFlux::Portal::Graphics::GpuBufferBinding::IMAGE_STORAGE, MayaFlux::Portal::Graphics::GpuBufferBinding::INPUT_OUTPUT, MayaFlux::is_image(), m_shared, MayaFlux::Portal::Graphics::GpuBufferBinding::OUTPUT, and unit_for().

Referenced by MayaFlux::Yantra::GpuDispatchCore::dispatch_core().

Here is the call graph for this function:

Here is the caller graph for this function: