mirror of
https://github.com/encounter/dawn-cmake.git
synced 2025-08-19 18:31:46 +00:00
305f4df2b5
This patch implements the struct FlatComputePipelineDescriptor to save all the pointer members of ComputePipelineDescriptor in CreateComputePipelineAsyncTask to better simplify the orignal code path. We will also implement FlatRenderPipelineDescriptor in the next patch, which is more important for CreateRenderPipelineAsync as there are many more pointer members in RenderPipelineDescriptor than those in ComputePipelineDescriptor. BUG=dawn:529 Change-Id: I69ab0dc898f0a6dcc9886d827729e6fd7308bd2d Reviewed-on: https://dawn-review.googlesource.com/c/dawn/+/63280 Reviewed-by: Austin Eng <enga@chromium.org> Commit-Queue: Jiawei Shao <jiawei.shao@intel.com>
101 lines
4.4 KiB
C++
101 lines
4.4 KiB
C++
// Copyright 2018 The Dawn Authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#include "dawn_native/vulkan/ComputePipelineVk.h"
|
|
|
|
#include "dawn_native/CreatePipelineAsyncTask.h"
|
|
#include "dawn_native/vulkan/DeviceVk.h"
|
|
#include "dawn_native/vulkan/FencedDeleter.h"
|
|
#include "dawn_native/vulkan/PipelineLayoutVk.h"
|
|
#include "dawn_native/vulkan/ShaderModuleVk.h"
|
|
#include "dawn_native/vulkan/UtilsVulkan.h"
|
|
#include "dawn_native/vulkan/VulkanError.h"
|
|
|
|
namespace dawn_native { namespace vulkan {
|
|
|
|
// static
|
|
ResultOrError<Ref<ComputePipeline>> ComputePipeline::Create(
|
|
Device* device,
|
|
const ComputePipelineDescriptor* descriptor) {
|
|
Ref<ComputePipeline> pipeline = AcquireRef(new ComputePipeline(device, descriptor));
|
|
DAWN_TRY(pipeline->Initialize(descriptor));
|
|
return pipeline;
|
|
}
|
|
|
|
MaybeError ComputePipeline::Initialize(const ComputePipelineDescriptor* descriptor) {
|
|
VkComputePipelineCreateInfo createInfo;
|
|
createInfo.sType = VK_STRUCTURE_TYPE_COMPUTE_PIPELINE_CREATE_INFO;
|
|
createInfo.pNext = nullptr;
|
|
createInfo.flags = 0;
|
|
createInfo.layout = ToBackend(descriptor->layout)->GetHandle();
|
|
createInfo.basePipelineHandle = ::VK_NULL_HANDLE;
|
|
createInfo.basePipelineIndex = -1;
|
|
|
|
createInfo.stage.sType = VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO;
|
|
createInfo.stage.pNext = nullptr;
|
|
createInfo.stage.flags = 0;
|
|
createInfo.stage.stage = VK_SHADER_STAGE_COMPUTE_BIT;
|
|
// Generate a new VkShaderModule with BindingRemapper tint transform for each pipeline
|
|
DAWN_TRY_ASSIGN(createInfo.stage.module,
|
|
ToBackend(descriptor->compute.module)
|
|
->GetTransformedModuleHandle(descriptor->compute.entryPoint,
|
|
ToBackend(GetLayout())));
|
|
createInfo.stage.pName = descriptor->compute.entryPoint;
|
|
createInfo.stage.pSpecializationInfo = nullptr;
|
|
|
|
Device* device = ToBackend(GetDevice());
|
|
|
|
PNextChainBuilder stageExtChain(&createInfo.stage);
|
|
|
|
VkPipelineShaderStageRequiredSubgroupSizeCreateInfoEXT subgroupSizeInfo = {};
|
|
uint32_t computeSubgroupSize = device->GetComputeSubgroupSize();
|
|
if (computeSubgroupSize != 0u) {
|
|
ASSERT(device->GetDeviceInfo().HasExt(DeviceExt::SubgroupSizeControl));
|
|
subgroupSizeInfo.requiredSubgroupSize = computeSubgroupSize;
|
|
stageExtChain.Add(
|
|
&subgroupSizeInfo,
|
|
VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_REQUIRED_SUBGROUP_SIZE_CREATE_INFO_EXT);
|
|
}
|
|
|
|
return CheckVkSuccess(
|
|
device->fn.CreateComputePipelines(device->GetVkDevice(), ::VK_NULL_HANDLE, 1,
|
|
&createInfo, nullptr, &*mHandle),
|
|
"CreateComputePipeline");
|
|
}
|
|
|
|
ComputePipeline::~ComputePipeline() {
|
|
if (mHandle != VK_NULL_HANDLE) {
|
|
ToBackend(GetDevice())->GetFencedDeleter()->DeleteWhenUnused(mHandle);
|
|
mHandle = VK_NULL_HANDLE;
|
|
}
|
|
}
|
|
|
|
VkPipeline ComputePipeline::GetHandle() const {
|
|
return mHandle;
|
|
}
|
|
|
|
void ComputePipeline::CreateAsync(Device* device,
|
|
std::unique_ptr<FlatComputePipelineDescriptor> descriptor,
|
|
size_t blueprintHash,
|
|
WGPUCreateComputePipelineAsyncCallback callback,
|
|
void* userdata) {
|
|
Ref<ComputePipeline> pipeline = AcquireRef(new ComputePipeline(device, descriptor.get()));
|
|
std::unique_ptr<CreateComputePipelineAsyncTask> asyncTask =
|
|
std::make_unique<CreateComputePipelineAsyncTask>(pipeline, std::move(descriptor),
|
|
blueprintHash, callback, userdata);
|
|
CreateComputePipelineAsyncTask::RunAsync(std::move(asyncTask));
|
|
}
|
|
|
|
}} // namespace dawn_native::vulkan
|