VulkanContext.hpp

Go to the documentation of this file.

 
#ifndef LLVMEXPR_RUNTIME_VULKAN_VULKANCONTEXT_HPP
#define LLVMEXPR_RUNTIME_VULKAN_VULKANCONTEXT_HPP
 
// NOLINTBEGIN(cppcoreguidelines-macro-usage,cppcoreguidelines-macro-to-enum,modernize-macro-to-enum)
 
#define VK_NO_PROTOTYPES
#define VULKAN_HPP_DISPATCH_LOADER_DYNAMIC 1
 
// NOLINTEND(cppcoreguidelines-macro-usage,cppcoreguidelines-macro-to-enum,modernize-macro-to-enum)
 
#include <mutex>
#include <vk_mem_alloc.h>
#include <vulkan/vulkan_raii.hpp>
 
namespace vkexpr {
 
class VulkanContext {
  public:
    static VulkanContext& getInstance();
    static VulkanContext& getInstance(int device_id);
 
    explicit VulkanContext(int device_id = -1);
    ~VulkanContext();
 
    VulkanContext(const VulkanContext&) = delete;
    VulkanContext& operator=(const VulkanContext&) = delete;
    VulkanContext(VulkanContext&&) = delete;
    VulkanContext& operator=(VulkanContext&&) = delete;
 
    vk::raii::Context& getContext() { return context; }
    vk::raii::Instance& getInstanceRef() { return instance; }
    vk::raii::PhysicalDevice& getPhysicalDevice() { return physical_device; }
    vk::raii::Device& getDevice() { return device; }
    vk::raii::Queue& getComputeQueue() { return compute_queue; }
    [[nodiscard]] uint32_t getQueueFamilyIndex() const {
        return queue_family_index;
    }
 
    void submit(const vk::SubmitInfo& submit_info, const vk::Fence& fence);
    void waitIdle();
 
  private:
    void createInstance();
    void pickPhysicalDevice();
    void createDevice();
 
    int device_id = -1;
    vk::raii::Context context;
    vk::raii::Instance instance;
    vk::raii::PhysicalDevice physical_device = nullptr;
    vk::raii::Device device = nullptr;
    vk::raii::Queue compute_queue = nullptr;
 
    uint32_t queue_family_index = -1;
    std::mutex queue_mutex;
};
 
} // namespace vkexpr
 
#endif // LLVMEXPR_RUNTIME_VULKAN_VULKANCONTEXT_HPP