1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
// Copyright 2019 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "third_party/blink/renderer/modules/webgpu/gpu_compute_pipeline.h"
#include "third_party/blink/renderer/bindings/modules/v8/v8_gpu_compute_pipeline_descriptor.h"
#include "third_party/blink/renderer/bindings/modules/v8/v8_gpu_programmable_stage.h"
#include "third_party/blink/renderer/modules/webgpu/gpu_bind_group_layout.h"
#include "third_party/blink/renderer/modules/webgpu/gpu_device.h"
#include "third_party/blink/renderer/modules/webgpu/gpu_pipeline_layout.h"
#include "third_party/blink/renderer/modules/webgpu/gpu_shader_module.h"
namespace blink {
WGPUComputePipelineDescriptor AsDawnType(
const GPUComputePipelineDescriptor* webgpu_desc,
std::string* label,
OwnedProgrammableStageDescriptor* computeStageDescriptor,
GPUDevice* device) {
DCHECK(webgpu_desc);
DCHECK(label);
DCHECK(computeStageDescriptor);
WGPUComputePipelineDescriptor dawn_desc = {};
dawn_desc.nextInChain = nullptr;
if (webgpu_desc->hasLayout()) {
dawn_desc.layout = AsDawnType(webgpu_desc->layout());
}
if (webgpu_desc->hasLabel()) {
*label = webgpu_desc->label().Utf8();
dawn_desc.label = label->c_str();
}
if (webgpu_desc->hasCompute()) {
*computeStageDescriptor = AsDawnType(webgpu_desc->compute());
} else if (webgpu_desc->hasComputeStage()) {
device->AddConsoleWarning(
"computeStage is deprecated. Use compute instead.");
*computeStageDescriptor = AsDawnType(webgpu_desc->computeStage());
}
dawn_desc.computeStage = std::get<0>(*computeStageDescriptor);
return dawn_desc;
}
// static
GPUComputePipeline* GPUComputePipeline::Create(
GPUDevice* device,
const GPUComputePipelineDescriptor* webgpu_desc) {
DCHECK(device);
DCHECK(webgpu_desc);
std::string label;
OwnedProgrammableStageDescriptor computeStageDescriptor;
WGPUComputePipelineDescriptor dawn_desc =
AsDawnType(webgpu_desc, &label, &computeStageDescriptor, device);
GPUComputePipeline* pipeline = MakeGarbageCollected<GPUComputePipeline>(
device, device->GetProcs().deviceCreateComputePipeline(
device->GetHandle(), &dawn_desc));
if (webgpu_desc->hasLabel())
pipeline->setLabel(webgpu_desc->label());
return pipeline;
}
GPUComputePipeline::GPUComputePipeline(GPUDevice* device,
WGPUComputePipeline compute_pipeline)
: DawnObject<WGPUComputePipeline>(device, compute_pipeline) {}
GPUBindGroupLayout* GPUComputePipeline::getBindGroupLayout(uint32_t index) {
return MakeGarbageCollected<GPUBindGroupLayout>(
device_,
GetProcs().computePipelineGetBindGroupLayout(GetHandle(), index));
}
} // namespace blink
|