vk_dispatch_node.hh

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2024 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */

 
#pragma once
 
#include "render_graph/vk_render_graph_links.hh"
#include "render_graph/vk_resource_access_info.hh"
#include "vk_node_info.hh"
 
namespace blender::gpu::render_graph {
struct VKDispatchData {
  VKPipelineData pipeline_data;
  uint32_t group_count_x;
  uint32_t group_count_y;
  uint32_t group_count_z;
};

struct VKDispatchCreateInfo : NonCopyable {
  VKDispatchData dispatch_node = {};
  const VKResourceAccessInfo &resources;
  VKDispatchCreateInfo(const VKResourceAccessInfo &resources) : resources(resources) {}
};
 
class VKDispatchNode : public VKNodeInfo<VKNodeType::DISPATCH,
                                         VKDispatchCreateInfo,
                                         VKDispatchData,
                                         VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT,
                                         VKResourceType::IMAGE | VKResourceType::BUFFER> {
 public:
  template<typename Node, typename Storage>
  static void set_node_data(Node &node, Storage & /* storage */, const CreateInfo &create_info)
  {
    node.dispatch = create_info.dispatch_node;
    vk_pipeline_data_copy(node.dispatch.pipeline_data, create_info.dispatch_node.pipeline_data);
  }

  void free_data(VKDispatchData &data)
  {
    vk_pipeline_data_free(data.pipeline_data);
  }

  void build_links(VKResourceStateTracker &resources,
                   VKRenderGraphNodeLinks &node_links,
                   const CreateInfo &create_info) override
  {
    create_info.resources.build_links(resources, node_links);
  }

  void build_commands(VKCommandBufferInterface &command_buffer,
                      Data &data,
                      VKBoundPipelines &r_bound_pipelines) override
  {
    vk_pipeline_data_build_commands(command_buffer,
                                    data.pipeline_data,
                                    r_bound_pipelines.compute,
                                    VK_PIPELINE_BIND_POINT_COMPUTE,
                                    VK_SHADER_STAGE_COMPUTE_BIT);
    command_buffer.dispatch(data.group_count_x, data.group_count_y, data.group_count_z);
  }
};
}  // namespace blender::gpu::render_graph