vkexpr::VkExprExecutor::Impl Struct Reference

Collaboration diagram for vkexpr::VkExprExecutor::Impl:

Classes
struct	PlaneResources
struct	Stream

Public Member Functions
	Impl (int device_id, int num_streams, int num_inputs, std::array< std::vector< std::string >, 3 > glsl_stages, std::uint32_t num_props_floats)
int	acquireStreamIndex ()
void	releaseStreamIndex (int idx)
void	drain ()

Public Attributes
VulkanContext *	context = nullptr
int	num_inputs = 0
std::uint32_t	num_props_floats = 0
std::array< std::vector< std::string >, 3 >	glsl_stages
int	num_streams = 0
std::vector< std::unique_ptr< Stream > >	streams
std::counting_semaphore	semaphore {0}
std::queue< int >	free_stream_indices
std::mutex	stream_mutex

Detailed Description

Definition at line 272 of file VkExprExecutor.cpp.

Constructor & Destructor Documentation

Impl()

vkexpr::VkExprExecutor::Impl::Impl ( int device_id, int num_streams, int num_inputs, std::array< std::vector< std::string >, 3 > glsl_stages, std::uint32_t num_props_floats )

inline

Definition at line 349 of file VkExprExecutor.cpp.

        : context(&VulkanContext::getInstance(device_id)),
          num_inputs(num_inputs), num_props_floats(num_props_floats),
          glsl_stages(std::move(glsl_stages)), num_streams(num_streams),
          semaphore(num_streams) {
        [[unlikely]] if (num_inputs <= 0) {
            throw std::runtime_error("VkExprExecutor: num_inputs must be > 0.");
        }
        [[unlikely]] if (num_streams <= 0) {
            throw std::runtime_error(
                "VkExprExecutor: num_streams must be > 0.");
        }
 
        auto& ctx = *context;
        streams.resize(static_cast<size_t>(num_streams));
        for (int k = 0; k < num_streams; ++k) {
            streams[static_cast<size_t>(k)] = std::make_unique<Stream>();
            auto& stream = *streams[static_cast<size_t>(k)];
            stream.memory = std::make_unique<VulkanMemory>(ctx);
            free_stream_indices.push(k);
 
            vk::CommandPoolCreateInfo pool_info(
                vk::CommandPoolCreateFlagBits::eResetCommandBuffer,
                ctx.getQueueFamilyIndex());
            stream.command_pool =
                vk::raii::CommandPool(ctx.getDevice(), pool_info);
 
            vk::CommandBufferAllocateInfo cmd_info(
                *stream.command_pool, vk::CommandBufferLevel::ePrimary, 1);
            auto cmd_buffers =
                vk::raii::CommandBuffers(ctx.getDevice(), cmd_info);
            stream.command_buffer = std::move(cmd_buffers[0]);
 
            vk::FenceCreateInfo fence_info;
            stream.fence = vk::raii::Fence(ctx.getDevice(), fence_info);
 
            for (int plane = 0; plane < 3; ++plane) {
                if (this->glsl_stages.at(plane).empty()) {
                    continue;
                }
                auto& plane_pipelines = stream.pipelines.at(plane);
                auto& stages = this->glsl_stages.at(plane);
                plane_pipelines.resize(stages.size());
 
                for (size_t s = 0; s < stages.size(); ++s) {
                    plane_pipelines[s] =
                        std::make_unique<VulkanComputePipeline>(
                            ctx, stages[s],
                            static_cast<std::uint32_t>(num_inputs +
                                                       static_cast<int>(s)),
                            num_props_floats);
                }
            }
        }
    }

References context, free_stream_indices, glsl_stages, num_inputs, num_props_floats, num_streams, semaphore, and streams.

Member Function Documentation

acquireStreamIndex()

int vkexpr::VkExprExecutor::Impl::acquireStreamIndex ( )

inline

Definition at line 407 of file VkExprExecutor.cpp.

                             {
        semaphore.acquire();
        std::lock_guard<std::mutex> lock(stream_mutex);
        int idx = free_stream_indices.front();
        free_stream_indices.pop();
        return idx;
    }

References free_stream_indices, semaphore, and stream_mutex.

drain()

void vkexpr::VkExprExecutor::Impl::drain ( )

inline

Definition at line 423 of file VkExprExecutor.cpp.

                 {
        for (int i = 0; i < num_streams; ++i) {
            semaphore.acquire();
        }
    }

References num_streams, and semaphore.

releaseStreamIndex()

void vkexpr::VkExprExecutor::Impl::releaseStreamIndex ( int idx )

inline

Definition at line 415 of file VkExprExecutor.cpp.

                                     {
        {
            std::lock_guard<std::mutex> lock(stream_mutex);
            free_stream_indices.push(idx);
        }
        semaphore.release();
    }

References free_stream_indices, semaphore, and stream_mutex.

Member Data Documentation

context

VulkanContext* vkexpr::VkExprExecutor::Impl::context = nullptr

Definition at line 338 of file VkExprExecutor.cpp.

Referenced by Impl().

free_stream_indices

std::queue<int> vkexpr::VkExprExecutor::Impl::free_stream_indices

Definition at line 346 of file VkExprExecutor.cpp.

Referenced by acquireStreamIndex(), Impl(), and releaseStreamIndex().

glsl_stages

std::array<std::vector<std::string>, 3> vkexpr::VkExprExecutor::Impl::glsl_stages

Definition at line 341 of file VkExprExecutor.cpp.

Referenced by Impl().

num_inputs

int vkexpr::VkExprExecutor::Impl::num_inputs = 0

Definition at line 339 of file VkExprExecutor.cpp.

Referenced by Impl().

num_props_floats

std::uint32_t vkexpr::VkExprExecutor::Impl::num_props_floats = 0

Definition at line 340 of file VkExprExecutor.cpp.

Referenced by Impl().

num_streams

int vkexpr::VkExprExecutor::Impl::num_streams = 0

Definition at line 343 of file VkExprExecutor.cpp.

Referenced by drain(), and Impl().

semaphore

std::counting_semaphore vkexpr::VkExprExecutor::Impl::semaphore {0}

Definition at line 345 of file VkExprExecutor.cpp.

{0};

Referenced by acquireStreamIndex(), drain(), Impl(), and releaseStreamIndex().

stream_mutex

std::mutex vkexpr::VkExprExecutor::Impl::stream_mutex

Definition at line 347 of file VkExprExecutor.cpp.

Referenced by acquireStreamIndex(), and releaseStreamIndex().

streams

std::vector<std::unique_ptr<Stream> > vkexpr::VkExprExecutor::Impl::streams

Definition at line 344 of file VkExprExecutor.cpp.

Referenced by Impl().

---

The documentation for this struct was generated from the following file:

• llvmexpr/runtime/vulkan/VkExprExecutor.cpp