AronaCore/core/rhi/compute_pipeline.h
2024-02-20 14:09:44 +08:00

16 lines
524 B
C++

#pragma once
#include "pipeline.h"
class CORE_API compute_pipeline : public pipeline {
public:
~compute_pipeline() override;
void add_binding(uint32_t binding, vk::DescriptorType descriptor_type, uint32_t descriptor_count, vk::Sampler immutable_samplers) override;
void create() override;
void dispatch(uint32_t group_count_x, uint32_t group_count_y, uint32_t group_count_z) const;
void set_shader(const uint8_t *shader_code, size_t shader_code_size);
private:
vk::ShaderModule shader_module_;
};