| // Copyright 2025 The Dawn & Tint Authors |
| // |
| // Redistribution and use in source and binary forms, with or without |
| // modification, are permitted provided that the following conditions are met: |
| // |
| // 1. Redistributions of source code must retain the above copyright notice, this |
| // list of conditions and the following disclaimer. |
| // |
| // 2. Redistributions in binary form must reproduce the above copyright notice, |
| // this list of conditions and the following disclaimer in the documentation |
| // and/or other materials provided with the distribution. |
| // |
| // 3. Neither the name of the copyright holder nor the names of its |
| // contributors may be used to endorse or promote products derived from |
| // this software without specific prior written permission. |
| // |
| // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
| // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
| // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE |
| // DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE |
| // FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
| // DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR |
| // SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER |
| // CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, |
| // OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| |
| #include <limits> |
| #include <vector> |
| |
| #include "dawn/tests/DawnTest.h" |
| #include "dawn/utils/WGPUHelpers.h" |
| |
| namespace dawn { |
| namespace { |
| |
| class PackUnpack4x8NormTests : public DawnTest {}; |
| |
| TEST_P(PackUnpack4x8NormTests, Pack4x8Snorm) { |
| const char* computeShader = R"( |
| @group(0) @binding(0) var<storage, read_write> buf : array<u32>; |
| @group(0) @binding(1) var<storage, read> inputBuf : array<vec4f>; |
| |
| @compute @workgroup_size(1) |
| fn main() { |
| var r: vec2<u32>; |
| for (var i = 0; i < 8; i++) { |
| r.x = pack4x8snorm(inputBuf[i]); |
| buf[i] = r.x; |
| } |
| } |
| )"; |
| |
| static uint32_t kNumTests = 8; |
| |
| wgpu::BufferDescriptor bufferDesc; |
| bufferDesc.size = kNumTests * sizeof(uint32_t); |
| bufferDesc.usage = wgpu::BufferUsage::Storage | wgpu::BufferUsage::CopySrc; |
| wgpu::Buffer bufferOut = device.CreateBuffer(&bufferDesc); |
| |
| wgpu::Buffer inputBuffer = utils::CreateBufferFromData( |
| device, |
| wgpu::BufferUsage::CopySrc | wgpu::BufferUsage::CopyDst | wgpu::BufferUsage::Storage, |
| { |
| // clang-format off |
| 0.f, 0.f, 0.f, 0.f, // |
| 0.f, 0.f, 0.f, -1.f, // |
| 0.f, 0.f, 0.f, 1.f, // |
| 0.f, 0.f, -1.f, 0.f, // |
| 0.f, 1.f, 0.f, 0.f, // |
| -1.f, 0.f, 0.f, 0.f, // |
| 1.f, -1.f, 1.f, -1.f, // |
| std::numeric_limits<float>::max(), -0.495f, 0.5f, std::numeric_limits<float>::lowest(), |
| // clang-format on |
| }); |
| |
| wgpu::ComputePipelineDescriptor csDesc; |
| csDesc.compute.module = utils::CreateShaderModule(device, computeShader); |
| wgpu::ComputePipeline pipeline = device.CreateComputePipeline(&csDesc); |
| |
| wgpu::BindGroup bindGroup = utils::MakeBindGroup(device, pipeline.GetBindGroupLayout(0), |
| { |
| {0, bufferOut}, |
| {1, inputBuffer}, |
| }); |
| |
| wgpu::CommandEncoder encoder = device.CreateCommandEncoder(); |
| wgpu::ComputePassEncoder pass = encoder.BeginComputePass(); |
| pass.SetPipeline(pipeline); |
| pass.SetBindGroup(0, bindGroup); |
| pass.DispatchWorkgroups(1); |
| pass.End(); |
| wgpu::CommandBuffer commands = encoder.Finish(); |
| queue.Submit(1, &commands); |
| |
| uint32_t expected[] = {0u, 0x8100'0000u, 0x7f00'0000u, 0x0081'0000u, |
| 0x0000'7f00u, 0x0000'0081u, 0x817f'817fu, 0x8140'c17fu}; |
| EXPECT_BUFFER_U32_RANGE_EQ(reinterpret_cast<uint32_t*>(expected), bufferOut, 0, kNumTests); |
| } |
| |
| TEST_P(PackUnpack4x8NormTests, Pack4x8Unorm) { |
| const char* computeShader = R"( |
| @group(0) @binding(0) var<storage, read_write> buf : array<u32>; |
| @group(0) @binding(1) var<storage, read> inputBuf : array<vec4f>; |
| |
| @compute @workgroup_size(1) |
| fn main() { |
| var r: vec2<u32>; |
| for (var i = 0; i < 7; i++) { |
| r.x = pack4x8unorm(inputBuf[i]); |
| buf[i] = r.x; |
| } |
| } |
| )"; |
| |
| static uint32_t kNumTests = 7; |
| |
| wgpu::BufferDescriptor bufferDesc; |
| bufferDesc.size = kNumTests * sizeof(uint32_t); |
| bufferDesc.usage = wgpu::BufferUsage::Storage | wgpu::BufferUsage::CopySrc; |
| wgpu::Buffer bufferOut = device.CreateBuffer(&bufferDesc); |
| |
| wgpu::Buffer inputBuffer = utils::CreateBufferFromData( |
| device, |
| wgpu::BufferUsage::CopySrc | wgpu::BufferUsage::CopyDst | wgpu::BufferUsage::Storage, |
| { |
| // clang-format off |
| 0.f, 0.f, 0.f, 0.f, // |
| 0.f, 0.f, 0.f, 1.f, // |
| 0.f, 0.f, 1.f, 0.f, // |
| 0.f, 1.f, 0.f, 0.f, // |
| 1.f, 0.f, 0.f, 0.f, // |
| 1.f, 0.f, 1.f, 0.f, // |
| std::numeric_limits<float>::max(), 0.f, 0.5f, std::numeric_limits<float>::lowest(), |
| // clang-format on |
| }); |
| |
| wgpu::ComputePipelineDescriptor csDesc; |
| csDesc.compute.module = utils::CreateShaderModule(device, computeShader); |
| wgpu::ComputePipeline pipeline = device.CreateComputePipeline(&csDesc); |
| |
| wgpu::BindGroup bindGroup = utils::MakeBindGroup(device, pipeline.GetBindGroupLayout(0), |
| { |
| {0, bufferOut}, |
| {1, inputBuffer}, |
| }); |
| |
| wgpu::CommandEncoder encoder = device.CreateCommandEncoder(); |
| wgpu::ComputePassEncoder pass = encoder.BeginComputePass(); |
| pass.SetPipeline(pipeline); |
| pass.SetBindGroup(0, bindGroup); |
| pass.DispatchWorkgroups(1); |
| pass.End(); |
| wgpu::CommandBuffer commands = encoder.Finish(); |
| queue.Submit(1, &commands); |
| |
| uint32_t expected[] = {0u, 0xff00'0000u, 0x00ff'0000u, 0x0000'ff00u, |
| 0x0000'00ffu, 0x00ff'00ffu, 0x0080'00ffu}; |
| EXPECT_BUFFER_U32_RANGE_EQ(reinterpret_cast<uint32_t*>(expected), bufferOut, 0, kNumTests); |
| } |
| |
| TEST_P(PackUnpack4x8NormTests, Unpack4x8Unorm) { |
| const char* computeShader = R"( |
| @group(0) @binding(0) var<storage, read_write> buf : array<vec4f>; |
| @group(0) @binding(1) var<storage, read> inputBuf : array<u32>; |
| |
| @compute @workgroup_size(1) |
| fn main() { |
| var r: vec2<u32>; |
| for (var i = 0; i < 7; i++) { |
| r.x = inputBuf[i]; |
| buf[i] = unpack4x8unorm(r.x); |
| } |
| } |
| )"; |
| |
| static uint32_t kNumTests = 7; |
| |
| wgpu::BufferDescriptor bufferDesc; |
| bufferDesc.size = kNumTests * 4 * sizeof(float); |
| bufferDesc.usage = wgpu::BufferUsage::Storage | wgpu::BufferUsage::CopySrc; |
| wgpu::Buffer bufferOut = device.CreateBuffer(&bufferDesc); |
| |
| wgpu::Buffer inputBuffer = utils::CreateBufferFromData( |
| device, |
| wgpu::BufferUsage::CopySrc | wgpu::BufferUsage::CopyDst | wgpu::BufferUsage::Storage, |
| { |
| // clang-format off |
| 0x0000'0000u, |
| 0xff00'0000u, |
| 0x00ff'0000u, |
| 0x0000'ff00u, |
| 0x0000'00ffu, |
| 0x00ff'00ffu, |
| 0x0066'00ffu |
| // clang-format on |
| }); |
| |
| wgpu::ComputePipelineDescriptor csDesc; |
| csDesc.compute.module = utils::CreateShaderModule(device, computeShader); |
| wgpu::ComputePipeline pipeline = device.CreateComputePipeline(&csDesc); |
| |
| wgpu::BindGroup bindGroup = utils::MakeBindGroup(device, pipeline.GetBindGroupLayout(0), |
| { |
| {0, bufferOut}, |
| {1, inputBuffer}, |
| }); |
| |
| wgpu::CommandEncoder encoder = device.CreateCommandEncoder(); |
| wgpu::ComputePassEncoder pass = encoder.BeginComputePass(); |
| pass.SetPipeline(pipeline); |
| pass.SetBindGroup(0, bindGroup); |
| pass.DispatchWorkgroups(1); |
| pass.End(); |
| wgpu::CommandBuffer commands = encoder.Finish(); |
| queue.Submit(1, &commands); |
| |
| float expected[] = { |
| // clang-format off |
| 0.f, 0.f, 0.f, 0.f, |
| 0.f, 0.f, 0.f, 1.f, |
| 0.f, 0.f, 1.f, 0.f, |
| 0.f, 1.f, 0.f, 0.f, |
| 1.f, 0.f, 0.f, 0.f, |
| 1.f, 0.f, 1.f, 0.f, |
| 1.f, 0.f, 0.4f, 0.f |
| // clang-format on |
| }; |
| EXPECT_BUFFER_FLOAT_RANGE_EQ(expected, bufferOut, 0, kNumTests); |
| } |
| |
| TEST_P(PackUnpack4x8NormTests, Unpack4x8Snorm) { |
| const char* computeShader = R"( |
| @group(0) @binding(0) var<storage, read_write> buf : array<vec4f>; |
| @group(0) @binding(1) var<storage, read> inputBuf : array<u32>; |
| |
| @compute @workgroup_size(1) |
| fn main() { |
| var r: vec2<u32>; |
| for (var i = 0; i < 8; i++) { |
| r.x = inputBuf[i]; |
| buf[i] = unpack4x8snorm(r.x); |
| } |
| } |
| )"; |
| |
| static uint32_t kNumTests = 8; |
| |
| wgpu::BufferDescriptor bufferDesc; |
| bufferDesc.size = kNumTests * 4 * sizeof(float); |
| bufferDesc.usage = wgpu::BufferUsage::Storage | wgpu::BufferUsage::CopySrc; |
| wgpu::Buffer bufferOut = device.CreateBuffer(&bufferDesc); |
| |
| wgpu::Buffer inputBuffer = utils::CreateBufferFromData( |
| device, |
| wgpu::BufferUsage::CopySrc | wgpu::BufferUsage::CopyDst | wgpu::BufferUsage::Storage, |
| { |
| // clang-format off |
| 0x0000'0000u, |
| 0x8100'0000u, |
| 0x7f00'0000u, |
| 0x0081'0000u, |
| 0x0000'7f00u, |
| 0x0000'0081u, |
| 0x817f'817fu, |
| 0x816d'937fu |
| // clang-format on |
| }); |
| |
| wgpu::ComputePipelineDescriptor csDesc; |
| csDesc.compute.module = utils::CreateShaderModule(device, computeShader); |
| wgpu::ComputePipeline pipeline = device.CreateComputePipeline(&csDesc); |
| |
| wgpu::BindGroup bindGroup = utils::MakeBindGroup(device, pipeline.GetBindGroupLayout(0), |
| { |
| {0, bufferOut}, |
| {1, inputBuffer}, |
| }); |
| |
| wgpu::CommandEncoder encoder = device.CreateCommandEncoder(); |
| wgpu::ComputePassEncoder pass = encoder.BeginComputePass(); |
| pass.SetPipeline(pipeline); |
| pass.SetBindGroup(0, bindGroup); |
| pass.DispatchWorkgroups(1); |
| pass.End(); |
| wgpu::CommandBuffer commands = encoder.Finish(); |
| queue.Submit(1, &commands); |
| |
| float expected[] = { |
| // clang-format off |
| 0.f, 0.f, 0.f, 0.f, |
| 0.f, 0.f, 0.f, -1.f, |
| 0.f, 0.f, 0.f, 1.f, |
| 0.f, 0.f, -1.f, 0.f, |
| 0.f, 1.f, 0.f, 0.f, |
| -1.f, 0.f, 0.f, 0.f, |
| -1.f, 0.f, 1.f, 0.f, |
| 1.f, -1.f, 1.f, -1.f, |
| 1.f, -0.8582677165354f, 0.8582677165354f, -1.f |
| // clang-format on |
| }; |
| EXPECT_BUFFER_FLOAT_RANGE_EQ(expected, bufferOut, 0, kNumTests); |
| } |
| |
| // DawnTestBase::CreateDeviceImpl always enables allow_unsafe_apis toggle. |
| DAWN_INSTANTIATE_TEST(PackUnpack4x8NormTests, |
| VulkanBackend(), |
| VulkanBackend({"polyfill_pack_unpack_4x8_norm"}), |
| WebGPUBackend()); |
| |
| } // anonymous namespace |
| } // namespace dawn |
