12#include "flutter/fml/container.h"
13#include "flutter/fml/trace_event.h"
23#include "vulkan/vulkan_enums.hpp"
27PipelineLibraryVK::PipelineLibraryVK(
28 const std::shared_ptr<DeviceHolderVK>& device_holder,
29 std::shared_ptr<const Capabilities> caps,
31 std::shared_ptr<fml::ConcurrentTaskRunner> worker_task_runner)
32 : device_holder_(device_holder),
33 pso_cache_(
std::make_shared<PipelineCacheVK>(
std::move(caps),
35 std::move(cache_directory))),
36 worker_task_runner_(
std::move(worker_task_runner)) {
38 if (!pso_cache_->IsValid() || !worker_task_runner_) {
45PipelineLibraryVK::~PipelineLibraryVK() =
default;
48bool PipelineLibraryVK::IsValid()
const {
52std::unique_ptr<ComputePipelineVK> PipelineLibraryVK::CreateComputePipeline(
55 vk::ComputePipelineCreateInfo pipeline_info;
60 const auto entrypoint = desc.GetStageEntrypoint();
66 std::shared_ptr<DeviceHolderVK> strong_device = device_holder_.lock();
70 auto device_properties = strong_device->GetPhysicalDevice().getProperties();
71 auto max_wg_size = device_properties.limits.maxComputeWorkGroupSize;
75 vk::SpecializationMapEntry specialization_map_entry[1];
77 uint32_t workgroup_size_x = max_wg_size[0];
78 specialization_map_entry[0].constantID = 0;
79 specialization_map_entry[0].offset = 0;
80 specialization_map_entry[0].size =
sizeof(uint32_t);
82 vk::SpecializationInfo specialization_info;
83 specialization_info.mapEntryCount = 1;
84 specialization_info.pMapEntries = &specialization_map_entry[0];
85 specialization_info.dataSize =
sizeof(uint32_t);
86 specialization_info.pData = &workgroup_size_x;
88 vk::PipelineShaderStageCreateInfo
info;
89 info.setStage(vk::ShaderStageFlagBits::eCompute);
90 info.setPName(
"main");
91 info.setModule(ShaderFunctionVK::Cast(entrypoint.get())->GetModule());
92 info.setPSpecializationInfo(&specialization_info);
93 pipeline_info.setStage(
info);
98 std::vector<vk::DescriptorSetLayoutBinding> desc_bindings;
100 for (
auto layout :
desc.GetDescriptorSetLayouts()) {
102 desc_bindings.push_back(vk_desc_layout);
105 vk::DescriptorSetLayoutCreateInfo descs_layout_info;
106 descs_layout_info.setBindings(desc_bindings);
108 auto [descs_result, descs_layout] =
109 strong_device->GetDevice().createDescriptorSetLayoutUnique(
111 if (descs_result != vk::Result::eSuccess) {
116 ContextVK::SetDebugName(strong_device->GetDevice(), descs_layout.get(),
117 "Descriptor Set Layout " +
desc.GetLabel());
122 vk::PipelineLayoutCreateInfo pipeline_layout_info;
123 pipeline_layout_info.setSetLayouts(descs_layout.get());
124 auto pipeline_layout = strong_device->GetDevice().createPipelineLayoutUnique(
125 pipeline_layout_info);
126 if (pipeline_layout.result != vk::Result::eSuccess) {
127 VALIDATION_LOG <<
"Could not create pipeline layout for pipeline "
128 <<
desc.GetLabel() <<
": "
129 << vk::to_string(pipeline_layout.result);
132 pipeline_info.setLayout(pipeline_layout.value.get());
137 auto pipeline = pso_cache_->CreatePipeline(pipeline_info);
143 ContextVK::SetDebugName(strong_device->GetDevice(), *pipeline_layout.value,
144 "Pipeline Layout " +
desc.GetLabel());
145 ContextVK::SetDebugName(strong_device->GetDevice(), *pipeline,
146 "Pipeline " +
desc.GetLabel());
148 return std::make_unique<ComputePipelineVK>(
153 std::move(pipeline_layout.value),
154 std::move(descs_layout)
161 Lock lock(pipelines_mutex_);
162 if (
auto found = pipelines_.find(descriptor); found != pipelines_.end()) {
163 return found->second;
169 RealizedFuture<std::shared_ptr<Pipeline<PipelineDescriptor>>>(
nullptr)};
172 auto promise = std::make_shared<
174 auto pipeline_future =
176 pipelines_[descriptor] = pipeline_future;
178 auto weak_this = weak_from_this();
180 worker_task_runner_->PostTask([descriptor, weak_this, promise]() {
181 auto thiz = weak_this.lock();
183 promise->set_value(
nullptr);
184 VALIDATION_LOG <<
"Pipeline library was collected before the pipeline "
189 promise->set_value(PipelineVK::Create(
191 PipelineLibraryVK::Cast(*thiz).device_holder_.lock(),
196 return pipeline_future;
202 Lock lock(compute_pipelines_mutex_);
203 if (
auto found = compute_pipelines_.find(descriptor);
204 found != compute_pipelines_.end()) {
205 return found->second;
211 RealizedFuture<std::shared_ptr<Pipeline<ComputePipelineDescriptor>>>(
215 auto promise = std::make_shared<
216 std::promise<std::shared_ptr<Pipeline<ComputePipelineDescriptor>>>>();
218 descriptor, promise->get_future()};
219 compute_pipelines_[descriptor] = pipeline_future;
221 auto weak_this = weak_from_this();
223 worker_task_runner_->PostTask([descriptor, weak_this, promise]() {
224 auto self = weak_this.lock();
226 promise->set_value(
nullptr);
227 VALIDATION_LOG <<
"Pipeline library was collected before the pipeline "
233 PipelineLibraryVK::Cast(*self).CreateComputePipeline(descriptor);
235 promise->set_value(
nullptr);
240 promise->set_value(std::move(pipeline));
243 return pipeline_future;
247void PipelineLibraryVK::RemovePipelinesWithEntryPoint(
248 std::shared_ptr<const ShaderFunction>
function) {
249 Lock lock(pipelines_mutex_);
252 return item->first.GetEntrypointForStage(
function->GetStage())
257void PipelineLibraryVK::DidAcquireSurfaceFrame() {
258 if (++frames_acquired_ == 50u) {
259 PersistPipelineCacheToDisk();
263void PipelineLibraryVK::PersistPipelineCacheToDisk() {
264 worker_task_runner_->PostTask(
265 [weak_cache =
decltype(pso_cache_)::weak_type(pso_cache_)]() {
266 auto cache = weak_cache.lock();
270 cache->PersistCacheToDisk();
274const std::shared_ptr<PipelineCacheVK>& PipelineLibraryVK::GetPSOCache()
const {
278const std::shared_ptr<fml::ConcurrentTaskRunner>&
279PipelineLibraryVK::GetWorkerTaskRunner()
const {
280 return worker_task_runner_;
static void info(const char *fmt,...) SK_PRINTF_LIKE(1
const std::string & GetLabel() const
#define FML_DCHECK(condition)
Dart_NativeFunction function
void erase_if(Collection &container, std::function< bool(typename Collection::iterator)> predicate)
constexpr vk::DescriptorSetLayoutBinding ToVKDescriptorSetLayoutBinding(const DescriptorSetLayout &layout)
#define TRACE_EVENT0(category_group, name)