Add a basic compute shared memory test
Change-Id: I6fab17d59d58434d10def29923e71106605c4f32
Bug: None
Reviewed-on: https://dawn-review.googlesource.com/c/dawn/+/6580
Reviewed-by: Kai Ninomiya <kainino@chromium.org>
Reviewed-by: Austin Eng <enga@chromium.org>
Commit-Queue: Kai Ninomiya <kainino@chromium.org>
diff --git a/BUILD.gn b/BUILD.gn
index 4217394..b7b6a07 100644
--- a/BUILD.gn
+++ b/BUILD.gn
@@ -630,6 +630,7 @@
"src/tests/end2end/ClipSpaceTests.cpp",
"src/tests/end2end/ColorStateTests.cpp",
"src/tests/end2end/ComputeCopyStorageBufferTests.cpp",
+ "src/tests/end2end/ComputeSharedMemoryTests.cpp",
"src/tests/end2end/CopyTests.cpp",
"src/tests/end2end/DebugMarkerTests.cpp",
"src/tests/end2end/DepthStencilStateTests.cpp",
diff --git a/src/tests/end2end/ComputeSharedMemoryTests.cpp b/src/tests/end2end/ComputeSharedMemoryTests.cpp
new file mode 100644
index 0000000..4d18bbd
--- /dev/null
+++ b/src/tests/end2end/ComputeSharedMemoryTests.cpp
@@ -0,0 +1,114 @@
+// Copyright 2019 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 "tests/DawnTest.h"
+
+#include "utils/DawnHelpers.h"
+
+#include <array>
+
+class ComputeSharedMemoryTests : public DawnTest {
+ public:
+ static constexpr uint32_t kInstances = 11;
+
+ void BasicTest(const char* shader);
+};
+
+void ComputeSharedMemoryTests::BasicTest(const char* shader) {
+ auto bgl = utils::MakeBindGroupLayout(
+ device, {
+ {0, dawn::ShaderStageBit::Compute, dawn::BindingType::StorageBuffer},
+ });
+
+ // Set up shader and pipeline
+ auto module = utils::CreateShaderModule(device, dawn::ShaderStage::Compute, shader);
+ auto pl = utils::MakeBasicPipelineLayout(device, &bgl);
+
+ dawn::ComputePipelineDescriptor csDesc;
+ csDesc.layout = pl;
+
+ dawn::PipelineStageDescriptor computeStage;
+ computeStage.module = module;
+ computeStage.entryPoint = "main";
+ csDesc.computeStage = &computeStage;
+
+ dawn::ComputePipeline pipeline = device.CreateComputePipeline(&csDesc);
+
+ // Set up dst storage buffer
+ dawn::BufferDescriptor dstDesc;
+ dstDesc.size = sizeof(uint32_t);
+ dstDesc.usage = dawn::BufferUsageBit::Storage | dawn::BufferUsageBit::TransferSrc |
+ dawn::BufferUsageBit::TransferDst;
+ dawn::Buffer dst = device.CreateBuffer(&dstDesc);
+
+ const uint32_t zero = 0;
+ dst.SetSubData(0, sizeof(zero), reinterpret_cast<const uint8_t*>(&zero));
+
+ // Set up bind group and issue dispatch
+ dawn::BindGroup bindGroup = utils::MakeBindGroup(device, bgl, {
+ {0, dst, 0, sizeof(uint32_t)},
+ });
+
+ dawn::CommandBuffer commands;
+ {
+ dawn::CommandEncoder encoder = device.CreateCommandEncoder();
+ dawn::ComputePassEncoder pass = encoder.BeginComputePass();
+ pass.SetPipeline(pipeline);
+ pass.SetBindGroup(0, bindGroup, 0, nullptr);
+ pass.Dispatch(1, 1, 1);
+ pass.EndPass();
+
+ commands = encoder.Finish();
+ }
+
+ queue.Submit(1, &commands);
+
+ const uint32_t expected = kInstances;
+ EXPECT_BUFFER_U32_RANGE_EQ(&expected, dst, 0, 1);
+}
+
+// Basic shared memory test
+TEST_P(ComputeSharedMemoryTests, Basic) {
+ BasicTest(R"(
+ #version 450
+ const uint kTileSize = 4;
+ const uint kInstances = 11;
+
+ layout(local_size_x = kTileSize, local_size_y = kTileSize, local_size_z = 1) in;
+ layout(std140, set = 0, binding = 0) buffer Dst { uint x; } dst;
+ shared uint tmp;
+
+ void main() {
+ uint index = gl_LocalInvocationID.y * kTileSize + gl_LocalInvocationID.x;
+ if (index == 0) {
+ tmp = 0;
+ }
+ barrier();
+ for (uint i = 0; i < kInstances; ++i) {
+ if (i == index) {
+ tmp++;
+ }
+ barrier();
+ }
+ if (index == 0) {
+ dst.x = tmp;
+ }
+ })");
+}
+
+DAWN_INSTANTIATE_TEST(ComputeSharedMemoryTests,
+ D3D12Backend,
+ MetalBackend,
+ OpenGLBackend,
+ VulkanBackend);
