mirror of
https://github.com/KhronosGroup/SPIRV-Tools
synced 2024-11-30 06:50:06 +00:00
749 lines
27 KiB
C++
749 lines
27 KiB
C++
// Copyright (c) 2017 Google Inc.
|
|
//
|
|
// 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 <string>
|
|
#include <vector>
|
|
|
|
#include "gmock/gmock.h"
|
|
#include "spirv-tools/libspirv.hpp"
|
|
#include "spirv-tools/optimizer.hpp"
|
|
#include "test/opt/pass_fixture.h"
|
|
|
|
namespace spvtools {
|
|
namespace opt {
|
|
namespace {
|
|
|
|
using ::testing::Eq;
|
|
|
|
// Return a string that contains the minimum instructions needed to form
|
|
// a valid module. Other instructions can be appended to this string.
|
|
std::string Header() {
|
|
return R"(OpCapability Shader
|
|
OpCapability Linkage
|
|
OpMemoryModel Logical GLSL450
|
|
)";
|
|
}
|
|
|
|
TEST(Optimizer, CanRunNullPassWithDistinctInputOutputVectors) {
|
|
SpirvTools tools(SPV_ENV_UNIVERSAL_1_0);
|
|
std::vector<uint32_t> binary_in;
|
|
tools.Assemble(Header() + "OpName %foo \"foo\"\n%foo = OpTypeVoid",
|
|
&binary_in);
|
|
|
|
Optimizer opt(SPV_ENV_UNIVERSAL_1_0);
|
|
opt.RegisterPass(CreateNullPass());
|
|
std::vector<uint32_t> binary_out;
|
|
opt.Run(binary_in.data(), binary_in.size(), &binary_out);
|
|
|
|
std::string disassembly;
|
|
tools.Disassemble(binary_out.data(), binary_out.size(), &disassembly);
|
|
EXPECT_THAT(disassembly,
|
|
Eq(Header() + "OpName %foo \"foo\"\n%foo = OpTypeVoid\n"));
|
|
}
|
|
|
|
TEST(Optimizer, CanRunTransformingPassWithDistinctInputOutputVectors) {
|
|
SpirvTools tools(SPV_ENV_UNIVERSAL_1_0);
|
|
std::vector<uint32_t> binary_in;
|
|
tools.Assemble(Header() + "OpName %foo \"foo\"\n%foo = OpTypeVoid",
|
|
&binary_in);
|
|
|
|
Optimizer opt(SPV_ENV_UNIVERSAL_1_0);
|
|
opt.RegisterPass(CreateStripDebugInfoPass());
|
|
std::vector<uint32_t> binary_out;
|
|
opt.Run(binary_in.data(), binary_in.size(), &binary_out);
|
|
|
|
std::string disassembly;
|
|
tools.Disassemble(binary_out.data(), binary_out.size(), &disassembly);
|
|
EXPECT_THAT(disassembly, Eq(Header() + "%void = OpTypeVoid\n"));
|
|
}
|
|
|
|
TEST(Optimizer, CanRunNullPassWithAliasedVectors) {
|
|
SpirvTools tools(SPV_ENV_UNIVERSAL_1_0);
|
|
std::vector<uint32_t> binary;
|
|
tools.Assemble("OpName %foo \"foo\"\n%foo = OpTypeVoid", &binary);
|
|
|
|
Optimizer opt(SPV_ENV_UNIVERSAL_1_0);
|
|
opt.RegisterPass(CreateNullPass());
|
|
opt.Run(binary.data(), binary.size(), &binary); // This is the key.
|
|
|
|
std::string disassembly;
|
|
tools.Disassemble(binary.data(), binary.size(), &disassembly);
|
|
EXPECT_THAT(disassembly, Eq("OpName %foo \"foo\"\n%foo = OpTypeVoid\n"));
|
|
}
|
|
|
|
TEST(Optimizer, CanRunNullPassWithAliasedVectorDataButDifferentSize) {
|
|
SpirvTools tools(SPV_ENV_UNIVERSAL_1_0);
|
|
std::vector<uint32_t> binary;
|
|
tools.Assemble(Header() + "OpName %foo \"foo\"\n%foo = OpTypeVoid", &binary);
|
|
|
|
Optimizer opt(SPV_ENV_UNIVERSAL_1_0);
|
|
opt.RegisterPass(CreateNullPass());
|
|
auto orig_size = binary.size();
|
|
// Now change the size. Add a word that will be ignored
|
|
// by the optimizer.
|
|
binary.push_back(42);
|
|
EXPECT_THAT(orig_size + 1, Eq(binary.size()));
|
|
opt.Run(binary.data(), orig_size, &binary); // This is the key.
|
|
// The binary vector should have been rewritten.
|
|
EXPECT_THAT(binary.size(), Eq(orig_size));
|
|
|
|
std::string disassembly;
|
|
tools.Disassemble(binary.data(), binary.size(), &disassembly);
|
|
EXPECT_THAT(disassembly,
|
|
Eq(Header() + "OpName %foo \"foo\"\n%foo = OpTypeVoid\n"));
|
|
}
|
|
|
|
TEST(Optimizer, CanRunTransformingPassWithAliasedVectors) {
|
|
SpirvTools tools(SPV_ENV_UNIVERSAL_1_0);
|
|
std::vector<uint32_t> binary;
|
|
tools.Assemble(Header() + "OpName %foo \"foo\"\n%foo = OpTypeVoid", &binary);
|
|
|
|
Optimizer opt(SPV_ENV_UNIVERSAL_1_0);
|
|
opt.RegisterPass(CreateStripDebugInfoPass());
|
|
opt.Run(binary.data(), binary.size(), &binary); // This is the key
|
|
|
|
std::string disassembly;
|
|
tools.Disassemble(binary.data(), binary.size(), &disassembly);
|
|
EXPECT_THAT(disassembly, Eq(Header() + "%void = OpTypeVoid\n"));
|
|
}
|
|
|
|
TEST(Optimizer, CanValidateFlags) {
|
|
Optimizer opt(SPV_ENV_UNIVERSAL_1_0);
|
|
EXPECT_FALSE(opt.FlagHasValidForm("bad-flag"));
|
|
EXPECT_TRUE(opt.FlagHasValidForm("-O"));
|
|
EXPECT_TRUE(opt.FlagHasValidForm("-Os"));
|
|
EXPECT_FALSE(opt.FlagHasValidForm("-O2"));
|
|
EXPECT_TRUE(opt.FlagHasValidForm("--this_flag"));
|
|
}
|
|
|
|
TEST(Optimizer, CanRegisterPassesFromFlags) {
|
|
SpirvTools tools(SPV_ENV_UNIVERSAL_1_0);
|
|
Optimizer opt(SPV_ENV_UNIVERSAL_1_0);
|
|
|
|
spv_message_level_t msg_level;
|
|
const char* msg_fname;
|
|
spv_position_t msg_position;
|
|
const char* msg;
|
|
auto examine_message = [&msg_level, &msg_fname, &msg_position, &msg](
|
|
spv_message_level_t ml, const char* f,
|
|
const spv_position_t& p, const char* m) {
|
|
msg_level = ml;
|
|
msg_fname = f;
|
|
msg_position = p;
|
|
msg = m;
|
|
};
|
|
opt.SetMessageConsumer(examine_message);
|
|
|
|
std::vector<std::string> pass_flags = {
|
|
"--strip-debug",
|
|
"--strip-reflect",
|
|
"--set-spec-const-default-value=23:42 21:12",
|
|
"--if-conversion",
|
|
"--freeze-spec-const",
|
|
"--inline-entry-points-exhaustive",
|
|
"--inline-entry-points-opaque",
|
|
"--convert-local-access-chains",
|
|
"--eliminate-dead-code-aggressive",
|
|
"--eliminate-insert-extract",
|
|
"--eliminate-local-single-block",
|
|
"--eliminate-local-single-store",
|
|
"--merge-blocks",
|
|
"--merge-return",
|
|
"--eliminate-dead-branches",
|
|
"--eliminate-dead-functions",
|
|
"--eliminate-local-multi-store",
|
|
"--eliminate-dead-const",
|
|
"--eliminate-dead-inserts",
|
|
"--eliminate-dead-variables",
|
|
"--fold-spec-const-op-composite",
|
|
"--loop-unswitch",
|
|
"--scalar-replacement=300",
|
|
"--scalar-replacement",
|
|
"--strength-reduction",
|
|
"--unify-const",
|
|
"--flatten-decorations",
|
|
"--compact-ids",
|
|
"--cfg-cleanup",
|
|
"--local-redundancy-elimination",
|
|
"--loop-invariant-code-motion",
|
|
"--reduce-load-size",
|
|
"--redundancy-elimination",
|
|
"--private-to-local",
|
|
"--remove-duplicates",
|
|
"--workaround-1209",
|
|
"--replace-invalid-opcode",
|
|
"--simplify-instructions",
|
|
"--ssa-rewrite",
|
|
"--copy-propagate-arrays",
|
|
"--loop-fission=20",
|
|
"--loop-fusion=2",
|
|
"--loop-unroll",
|
|
"--vector-dce",
|
|
"--loop-unroll-partial=3",
|
|
"--loop-peeling",
|
|
"--ccp",
|
|
"-O",
|
|
"-Os",
|
|
"--legalize-hlsl"};
|
|
EXPECT_TRUE(opt.RegisterPassesFromFlags(pass_flags));
|
|
|
|
// Test some invalid flags.
|
|
EXPECT_FALSE(opt.RegisterPassFromFlag("-O2"));
|
|
EXPECT_EQ(msg_level, SPV_MSG_ERROR);
|
|
|
|
EXPECT_FALSE(opt.RegisterPassFromFlag("-loop-unroll"));
|
|
EXPECT_EQ(msg_level, SPV_MSG_ERROR);
|
|
|
|
EXPECT_FALSE(opt.RegisterPassFromFlag("--set-spec-const-default-value"));
|
|
EXPECT_EQ(msg_level, SPV_MSG_ERROR);
|
|
|
|
EXPECT_FALSE(opt.RegisterPassFromFlag("--scalar-replacement=s"));
|
|
EXPECT_EQ(msg_level, SPV_MSG_ERROR);
|
|
|
|
EXPECT_FALSE(opt.RegisterPassFromFlag("--loop-fission=-4"));
|
|
EXPECT_EQ(msg_level, SPV_MSG_ERROR);
|
|
|
|
EXPECT_FALSE(opt.RegisterPassFromFlag("--loop-fusion=xx"));
|
|
EXPECT_EQ(msg_level, SPV_MSG_ERROR);
|
|
|
|
EXPECT_FALSE(opt.RegisterPassFromFlag("--loop-unroll-partial"));
|
|
EXPECT_EQ(msg_level, SPV_MSG_ERROR);
|
|
}
|
|
|
|
TEST(Optimizer, VulkanToWebGPUSetsCorrectPasses) {
|
|
Optimizer opt(SPV_ENV_WEBGPU_0);
|
|
opt.RegisterVulkanToWebGPUPasses();
|
|
std::vector<const char*> pass_names = opt.GetPassNames();
|
|
|
|
std::vector<std::string> registered_passes;
|
|
for (auto name = pass_names.begin(); name != pass_names.end(); ++name)
|
|
registered_passes.push_back(*name);
|
|
|
|
std::vector<std::string> expected_passes = {"eliminate-dead-branches",
|
|
"eliminate-dead-code-aggressive",
|
|
"eliminate-dead-const",
|
|
"flatten-decorations",
|
|
"strip-debug",
|
|
"strip-atomic-counter-memory",
|
|
"generate-webgpu-initializers",
|
|
"legalize-vector-shuffle",
|
|
"split-invalid-unreachable",
|
|
"compact-ids"};
|
|
std::sort(registered_passes.begin(), registered_passes.end());
|
|
std::sort(expected_passes.begin(), expected_passes.end());
|
|
|
|
ASSERT_EQ(registered_passes.size(), expected_passes.size());
|
|
for (size_t i = 0; i < registered_passes.size(); i++)
|
|
EXPECT_EQ(registered_passes[i], expected_passes[i]);
|
|
}
|
|
|
|
struct VulkanToWebGPUPassCase {
|
|
// Input SPIR-V
|
|
std::string input;
|
|
// Expected result SPIR-V
|
|
std::string expected;
|
|
// Specific pass under test, used for logging messages.
|
|
std::string pass;
|
|
};
|
|
|
|
using VulkanToWebGPUPassTest =
|
|
PassTest<::testing::TestWithParam<VulkanToWebGPUPassCase>>;
|
|
|
|
TEST_P(VulkanToWebGPUPassTest, Ran) {
|
|
std::vector<uint32_t> binary;
|
|
{
|
|
SpirvTools tools(SPV_ENV_VULKAN_1_1);
|
|
tools.Assemble(GetParam().input, &binary);
|
|
}
|
|
|
|
Optimizer opt(SPV_ENV_WEBGPU_0);
|
|
opt.RegisterVulkanToWebGPUPasses();
|
|
|
|
std::vector<uint32_t> optimized;
|
|
class ValidatorOptions validator_options;
|
|
ASSERT_TRUE(opt.Run(binary.data(), binary.size(), &optimized,
|
|
validator_options, true));
|
|
std::string disassembly;
|
|
{
|
|
SpirvTools tools(SPV_ENV_WEBGPU_0);
|
|
tools.Disassemble(optimized.data(), optimized.size(), &disassembly);
|
|
}
|
|
|
|
EXPECT_EQ(GetParam().expected, disassembly)
|
|
<< "Was expecting pass '" << GetParam().pass << "' to have been run.\n";
|
|
}
|
|
|
|
INSTANTIATE_TEST_SUITE_P(
|
|
Optimizer, VulkanToWebGPUPassTest,
|
|
::testing::ValuesIn(std::vector<VulkanToWebGPUPassCase>{
|
|
// FlattenDecorations
|
|
{// input
|
|
"OpCapability Shader\n"
|
|
"OpCapability VulkanMemoryModelKHR\n"
|
|
"OpExtension \"SPV_KHR_vulkan_memory_model\"\n"
|
|
"OpMemoryModel Logical VulkanKHR\n"
|
|
"OpEntryPoint Fragment %main \"main\" %hue %saturation %value\n"
|
|
"OpExecutionMode %main OriginUpperLeft\n"
|
|
"OpDecorate %group Flat\n"
|
|
"OpDecorate %group NoPerspective\n"
|
|
"%group = OpDecorationGroup\n"
|
|
"%void = OpTypeVoid\n"
|
|
"%void_fn = OpTypeFunction %void\n"
|
|
"%float = OpTypeFloat 32\n"
|
|
"%_ptr_Input_float = OpTypePointer Input %float\n"
|
|
"%hue = OpVariable %_ptr_Input_float Input\n"
|
|
"%saturation = OpVariable %_ptr_Input_float Input\n"
|
|
"%value = OpVariable %_ptr_Input_float Input\n"
|
|
"%main = OpFunction %void None %void_fn\n"
|
|
"%entry = OpLabel\n"
|
|
"OpReturn\n"
|
|
"OpFunctionEnd\n",
|
|
// expected
|
|
"OpCapability Shader\n"
|
|
"OpCapability VulkanMemoryModelKHR\n"
|
|
"OpExtension \"SPV_KHR_vulkan_memory_model\"\n"
|
|
"OpMemoryModel Logical VulkanKHR\n"
|
|
"OpEntryPoint Fragment %1 \"main\" %2 %3 %4\n"
|
|
"OpExecutionMode %1 OriginUpperLeft\n"
|
|
"%void = OpTypeVoid\n"
|
|
"%6 = OpTypeFunction %void\n"
|
|
"%float = OpTypeFloat 32\n"
|
|
"%_ptr_Input_float = OpTypePointer Input %float\n"
|
|
"%2 = OpVariable %_ptr_Input_float Input\n"
|
|
"%3 = OpVariable %_ptr_Input_float Input\n"
|
|
"%4 = OpVariable %_ptr_Input_float Input\n"
|
|
"%1 = OpFunction %void None %6\n"
|
|
"%9 = OpLabel\n"
|
|
"OpReturn\n"
|
|
"OpFunctionEnd\n",
|
|
// pass
|
|
"flatten-decorations"},
|
|
// Strip Debug
|
|
{// input
|
|
"OpCapability Shader\n"
|
|
"OpCapability VulkanMemoryModelKHR\n"
|
|
"OpExtension \"SPV_KHR_vulkan_memory_model\"\n"
|
|
"OpMemoryModel Logical VulkanKHR\n"
|
|
"OpEntryPoint Vertex %func \"shader\"\n"
|
|
"OpName %main \"main\"\n"
|
|
"OpName %void_fn \"void_fn\"\n"
|
|
"%void = OpTypeVoid\n"
|
|
"%void_f = OpTypeFunction %void\n"
|
|
"%func = OpFunction %void None %void_f\n"
|
|
"%label = OpLabel\n"
|
|
"OpReturn\n"
|
|
"OpFunctionEnd\n",
|
|
// expected
|
|
"OpCapability Shader\n"
|
|
"OpCapability VulkanMemoryModelKHR\n"
|
|
"OpExtension \"SPV_KHR_vulkan_memory_model\"\n"
|
|
"OpMemoryModel Logical VulkanKHR\n"
|
|
"OpEntryPoint Vertex %1 \"shader\"\n"
|
|
"%void = OpTypeVoid\n"
|
|
"%3 = OpTypeFunction %void\n"
|
|
"%1 = OpFunction %void None %3\n"
|
|
"%4 = OpLabel\n"
|
|
"OpReturn\n"
|
|
"OpFunctionEnd\n",
|
|
// pass
|
|
"strip-debug"},
|
|
// Eliminate Dead Constants
|
|
{// input
|
|
"OpCapability Shader\n"
|
|
"OpCapability VulkanMemoryModelKHR\n"
|
|
"OpExtension \"SPV_KHR_vulkan_memory_model\"\n"
|
|
"OpMemoryModel Logical VulkanKHR\n"
|
|
"OpEntryPoint Vertex %func \"shader\"\n"
|
|
"%u32 = OpTypeInt 32 0\n"
|
|
"%u32_ptr = OpTypePointer Workgroup %u32\n"
|
|
"%u32_var = OpVariable %u32_ptr Workgroup\n"
|
|
"%u32_1 = OpConstant %u32 1\n"
|
|
"%cross_device = OpConstant %u32 0\n"
|
|
"%relaxed = OpConstant %u32 0\n"
|
|
"%acquire_release_atomic_counter_workgroup = OpConstant %u32 1288\n"
|
|
"%void = OpTypeVoid\n"
|
|
"%void_f = OpTypeFunction %void\n"
|
|
"%func = OpFunction %void None %void_f\n"
|
|
"%label = OpLabel\n"
|
|
"OpReturn\n"
|
|
"OpFunctionEnd\n",
|
|
// expected
|
|
"OpCapability Shader\n"
|
|
"OpCapability VulkanMemoryModelKHR\n"
|
|
"OpExtension \"SPV_KHR_vulkan_memory_model\"\n"
|
|
"OpMemoryModel Logical VulkanKHR\n"
|
|
"OpEntryPoint Vertex %1 \"shader\"\n"
|
|
"%uint = OpTypeInt 32 0\n"
|
|
"%_ptr_Workgroup_uint = OpTypePointer Workgroup %uint\n"
|
|
"%4 = OpVariable %_ptr_Workgroup_uint Workgroup\n"
|
|
"%void = OpTypeVoid\n"
|
|
"%6 = OpTypeFunction %void\n"
|
|
"%1 = OpFunction %void None %6\n"
|
|
"%7 = OpLabel\n"
|
|
"OpReturn\n"
|
|
"OpFunctionEnd\n",
|
|
"eliminate-dead-const"},
|
|
// Strip Atomic Counter Memory
|
|
{// input
|
|
"OpCapability Shader\n"
|
|
"OpCapability VulkanMemoryModelKHR\n"
|
|
"OpExtension \"SPV_KHR_vulkan_memory_model\"\n"
|
|
"OpMemoryModel Logical VulkanKHR\n"
|
|
"OpEntryPoint Vertex %func \"shader\"\n"
|
|
"%u32 = OpTypeInt 32 0\n"
|
|
"%u32_ptr = OpTypePointer Workgroup %u32\n"
|
|
"%u32_var = OpVariable %u32_ptr Workgroup\n"
|
|
"%u32_0 = OpConstant %u32 0\n"
|
|
"%u32_1 = OpConstant %u32 1\n"
|
|
"%cross_device = OpConstant %u32 0\n"
|
|
"%acquire_release_atomic_counter_workgroup = OpConstant %u32 1288\n"
|
|
"%void = OpTypeVoid\n"
|
|
"%void_f = OpTypeFunction %void\n"
|
|
"%func = OpFunction %void None %void_f\n"
|
|
"%label = OpLabel\n"
|
|
"%val0 = OpAtomicStore %u32_var %cross_device "
|
|
"%acquire_release_atomic_counter_workgroup %u32_1\n"
|
|
"%val1 = OpAtomicIIncrement %u32 %u32_var %cross_device "
|
|
"%acquire_release_atomic_counter_workgroup\n"
|
|
"%val2 = OpAtomicCompareExchange %u32 %u32_var %cross_device "
|
|
"%acquire_release_atomic_counter_workgroup "
|
|
"%acquire_release_atomic_counter_workgroup %u32_0 %u32_0\n"
|
|
"OpReturn\n"
|
|
"OpFunctionEnd\n",
|
|
// expected
|
|
"OpCapability Shader\n"
|
|
"OpCapability VulkanMemoryModelKHR\n"
|
|
"OpExtension \"SPV_KHR_vulkan_memory_model\"\n"
|
|
"OpMemoryModel Logical VulkanKHR\n"
|
|
"OpEntryPoint Vertex %1 \"shader\"\n"
|
|
"%uint = OpTypeInt 32 0\n"
|
|
"%_ptr_Workgroup_uint = OpTypePointer Workgroup %uint\n"
|
|
"%4 = OpVariable %_ptr_Workgroup_uint Workgroup\n"
|
|
"%uint_0 = OpConstant %uint 0\n"
|
|
"%uint_1 = OpConstant %uint 1\n"
|
|
"%uint_0_0 = OpConstant %uint 0\n"
|
|
"%void = OpTypeVoid\n"
|
|
"%9 = OpTypeFunction %void\n"
|
|
"%uint_264 = OpConstant %uint 264\n"
|
|
"%1 = OpFunction %void None %9\n"
|
|
"%11 = OpLabel\n"
|
|
"OpAtomicStore %4 %uint_0_0 %uint_264 %uint_1\n"
|
|
"%12 = OpAtomicIIncrement %uint %4 %uint_0_0 %uint_264\n"
|
|
"%13 = OpAtomicCompareExchange %uint %4 %uint_0_0 %uint_264 %uint_264 "
|
|
"%uint_0 %uint_0\n"
|
|
"OpReturn\n"
|
|
"OpFunctionEnd\n",
|
|
// pass
|
|
"strip-atomic-counter-memory"},
|
|
// Generate WebGPU Initializers
|
|
{// input
|
|
"OpCapability Shader\n"
|
|
"OpCapability VulkanMemoryModelKHR\n"
|
|
"OpExtension \"SPV_KHR_vulkan_memory_model\"\n"
|
|
"OpMemoryModel Logical VulkanKHR\n"
|
|
"OpEntryPoint Vertex %func \"shader\"\n"
|
|
"%u32 = OpTypeInt 32 0\n"
|
|
"%u32_ptr = OpTypePointer Private %u32\n"
|
|
"%u32_var = OpVariable %u32_ptr Private\n"
|
|
"%u32_0 = OpConstant %u32 0\n"
|
|
"%void = OpTypeVoid\n"
|
|
"%void_f = OpTypeFunction %void\n"
|
|
"%func = OpFunction %void None %void_f\n"
|
|
"%label = OpLabel\n"
|
|
"OpStore %u32_var %u32_0\n"
|
|
"OpReturn\n"
|
|
"OpFunctionEnd\n",
|
|
// expected
|
|
"OpCapability Shader\n"
|
|
"OpCapability VulkanMemoryModelKHR\n"
|
|
"OpExtension \"SPV_KHR_vulkan_memory_model\"\n"
|
|
"OpMemoryModel Logical VulkanKHR\n"
|
|
"OpEntryPoint Vertex %1 \"shader\"\n"
|
|
"%uint = OpTypeInt 32 0\n"
|
|
"%_ptr_Private_uint = OpTypePointer Private %uint\n"
|
|
"%4 = OpConstantNull %uint\n"
|
|
"%5 = OpVariable %_ptr_Private_uint Private %4\n"
|
|
"%uint_0 = OpConstant %uint 0\n"
|
|
"%void = OpTypeVoid\n"
|
|
"%8 = OpTypeFunction %void\n"
|
|
"%1 = OpFunction %void None %8\n"
|
|
"%9 = OpLabel\n"
|
|
"OpStore %5 %uint_0\n"
|
|
"OpReturn\n"
|
|
"OpFunctionEnd\n",
|
|
// pass
|
|
"generate-webgpu-initializers"},
|
|
// Legalize Vector Shuffle
|
|
{// input
|
|
"OpCapability Shader\n"
|
|
"OpCapability VulkanMemoryModelKHR\n"
|
|
"OpExtension \"SPV_KHR_vulkan_memory_model\"\n"
|
|
"OpMemoryModel Logical VulkanKHR\n"
|
|
"OpEntryPoint Vertex %1 \"shader\"\n"
|
|
"%uint = OpTypeInt 32 0\n"
|
|
"%v3uint = OpTypeVector %uint 3\n"
|
|
"%_ptr_Function_v3uint = OpTypePointer Function %v3uint\n"
|
|
"%void = OpTypeVoid\n"
|
|
"%6 = OpTypeFunction %void\n"
|
|
"%1 = OpFunction %void None %6\n"
|
|
"%7 = OpLabel\n"
|
|
"%8 = OpVariable %_ptr_Function_v3uint Function\n"
|
|
"%9 = OpLoad %v3uint %8\n"
|
|
"%10 = OpLoad %v3uint %8\n"
|
|
"%11 = OpVectorShuffle %v3uint %9 %10 2 1 0xFFFFFFFF\n"
|
|
"OpReturn\n"
|
|
"OpFunctionEnd\n",
|
|
// expected
|
|
"OpCapability Shader\n"
|
|
"OpCapability VulkanMemoryModelKHR\n"
|
|
"OpExtension \"SPV_KHR_vulkan_memory_model\"\n"
|
|
"OpMemoryModel Logical VulkanKHR\n"
|
|
"OpEntryPoint Vertex %1 \"shader\"\n"
|
|
"%uint = OpTypeInt 32 0\n"
|
|
"%v3uint = OpTypeVector %uint 3\n"
|
|
"%_ptr_Function_v3uint = OpTypePointer Function %v3uint\n"
|
|
"%void = OpTypeVoid\n"
|
|
"%6 = OpTypeFunction %void\n"
|
|
"%7 = OpConstantNull %v3uint\n"
|
|
"%1 = OpFunction %void None %6\n"
|
|
"%8 = OpLabel\n"
|
|
"%9 = OpVariable %_ptr_Function_v3uint Function %7\n"
|
|
"%10 = OpLoad %v3uint %9\n"
|
|
"%11 = OpLoad %v3uint %9\n"
|
|
"%12 = OpVectorShuffle %v3uint %10 %11 2 1 0\n"
|
|
"OpReturn\n"
|
|
"OpFunctionEnd\n",
|
|
// pass
|
|
"legalize-vector-shuffle"},
|
|
// Split Invalid Unreachable
|
|
{// input
|
|
"OpCapability Shader\n"
|
|
"OpCapability VulkanMemoryModelKHR\n"
|
|
"OpExtension \"SPV_KHR_vulkan_memory_model\"\n"
|
|
"OpMemoryModel Logical VulkanKHR\n"
|
|
"OpEntryPoint Vertex %1 \"shader\"\n"
|
|
"%uint = OpTypeInt 32 0\n"
|
|
"%uint_1 = OpConstant %uint 1\n"
|
|
"%uint_2 = OpConstant %uint 2\n"
|
|
"%void = OpTypeVoid\n"
|
|
"%bool = OpTypeBool\n"
|
|
"%7 = OpTypeFunction %void\n"
|
|
"%1 = OpFunction %void None %7\n"
|
|
"%8 = OpLabel\n"
|
|
"OpBranch %9\n"
|
|
"%9 = OpLabel\n"
|
|
"OpLoopMerge %10 %11 None\n"
|
|
"OpBranch %12\n"
|
|
"%12 = OpLabel\n"
|
|
"%13 = OpSLessThan %bool %uint_1 %uint_2\n"
|
|
"OpSelectionMerge %11 None\n"
|
|
"OpBranchConditional %13 %14 %15\n"
|
|
"%14 = OpLabel\n"
|
|
"OpReturn\n"
|
|
"%15 = OpLabel\n"
|
|
"OpReturn\n"
|
|
"%10 = OpLabel\n"
|
|
"OpUnreachable\n"
|
|
"%11 = OpLabel\n"
|
|
"OpBranch %9\n"
|
|
"OpFunctionEnd\n",
|
|
// expected
|
|
"OpCapability Shader\n"
|
|
"OpCapability VulkanMemoryModelKHR\n"
|
|
"OpExtension \"SPV_KHR_vulkan_memory_model\"\n"
|
|
"OpMemoryModel Logical VulkanKHR\n"
|
|
"OpEntryPoint Vertex %1 \"shader\"\n"
|
|
"%uint = OpTypeInt 32 0\n"
|
|
"%uint_1 = OpConstant %uint 1\n"
|
|
"%uint_2 = OpConstant %uint 2\n"
|
|
"%void = OpTypeVoid\n"
|
|
"%bool = OpTypeBool\n"
|
|
"%7 = OpTypeFunction %void\n"
|
|
"%1 = OpFunction %void None %7\n"
|
|
"%8 = OpLabel\n"
|
|
"OpBranch %9\n"
|
|
"%9 = OpLabel\n"
|
|
"OpLoopMerge %10 %11 None\n"
|
|
"OpBranch %12\n"
|
|
"%12 = OpLabel\n"
|
|
"%13 = OpSLessThan %bool %uint_1 %uint_2\n"
|
|
"OpSelectionMerge %14 None\n"
|
|
"OpBranchConditional %13 %15 %16\n"
|
|
"%15 = OpLabel\n"
|
|
"OpReturn\n"
|
|
"%16 = OpLabel\n"
|
|
"OpReturn\n"
|
|
"%10 = OpLabel\n"
|
|
"OpUnreachable\n"
|
|
"%14 = OpLabel\n"
|
|
"OpUnreachable\n"
|
|
"%11 = OpLabel\n"
|
|
"OpBranch %9\n"
|
|
"OpFunctionEnd\n",
|
|
// pass
|
|
"split-invalid-unreachable"},
|
|
// Compact IDs
|
|
{// input
|
|
"OpCapability Shader\n"
|
|
"OpCapability VulkanMemoryModelKHR\n"
|
|
"OpExtension \"SPV_KHR_vulkan_memory_model\"\n"
|
|
"OpMemoryModel Logical VulkanKHR\n"
|
|
"OpEntryPoint Vertex %1000 \"shader\"\n"
|
|
"%10 = OpTypeVoid\n"
|
|
"%100 = OpTypeFunction %10\n"
|
|
"%1000 = OpFunction %10 None %100\n"
|
|
"%10000 = OpLabel\n"
|
|
"OpReturn\n"
|
|
"OpFunctionEnd\n",
|
|
// expected
|
|
"OpCapability Shader\n"
|
|
"OpCapability VulkanMemoryModelKHR\n"
|
|
"OpExtension \"SPV_KHR_vulkan_memory_model\"\n"
|
|
"OpMemoryModel Logical VulkanKHR\n"
|
|
"OpEntryPoint Vertex %1 \"shader\"\n"
|
|
"%void = OpTypeVoid\n"
|
|
"%3 = OpTypeFunction %void\n"
|
|
"%1 = OpFunction %void None %3\n"
|
|
"%4 = OpLabel\n"
|
|
"OpReturn\n"
|
|
"OpFunctionEnd\n",
|
|
// pass
|
|
"compact-ids"}}));
|
|
|
|
TEST(Optimizer, WebGPUToVulkanSetsCorrectPasses) {
|
|
Optimizer opt(SPV_ENV_VULKAN_1_1);
|
|
opt.RegisterWebGPUToVulkanPasses();
|
|
std::vector<const char*> pass_names = opt.GetPassNames();
|
|
|
|
std::vector<std::string> registered_passes;
|
|
for (auto name = pass_names.begin(); name != pass_names.end(); ++name)
|
|
registered_passes.push_back(*name);
|
|
|
|
std::vector<std::string> expected_passes = {"decompose-initialized-variables",
|
|
"compact-ids"};
|
|
std::sort(registered_passes.begin(), registered_passes.end());
|
|
std::sort(expected_passes.begin(), expected_passes.end());
|
|
|
|
ASSERT_EQ(registered_passes.size(), expected_passes.size());
|
|
for (size_t i = 0; i < registered_passes.size(); i++)
|
|
EXPECT_EQ(registered_passes[i], expected_passes[i]);
|
|
}
|
|
|
|
struct WebGPUToVulkanPassCase {
|
|
// Input SPIR-V
|
|
std::string input;
|
|
// Expected result SPIR-V
|
|
std::string expected;
|
|
// Specific pass under test, used for logging messages.
|
|
std::string pass;
|
|
};
|
|
|
|
using WebGPUToVulkanPassTest =
|
|
PassTest<::testing::TestWithParam<WebGPUToVulkanPassCase>>;
|
|
|
|
TEST_P(WebGPUToVulkanPassTest, Ran) {
|
|
std::vector<uint32_t> binary;
|
|
{
|
|
SpirvTools tools(SPV_ENV_WEBGPU_0);
|
|
tools.Assemble(GetParam().input, &binary);
|
|
}
|
|
|
|
Optimizer opt(SPV_ENV_VULKAN_1_1);
|
|
opt.RegisterWebGPUToVulkanPasses();
|
|
|
|
std::vector<uint32_t> optimized;
|
|
class ValidatorOptions validator_options;
|
|
ASSERT_TRUE(opt.Run(binary.data(), binary.size(), &optimized,
|
|
validator_options, true));
|
|
std::string disassembly;
|
|
{
|
|
SpirvTools tools(SPV_ENV_VULKAN_1_1);
|
|
tools.Disassemble(optimized.data(), optimized.size(), &disassembly);
|
|
}
|
|
|
|
EXPECT_EQ(GetParam().expected, disassembly)
|
|
<< "Was expecting pass '" << GetParam().pass << "' to have been run.\n";
|
|
}
|
|
|
|
INSTANTIATE_TEST_SUITE_P(
|
|
Optimizer, WebGPUToVulkanPassTest,
|
|
::testing::ValuesIn(std::vector<WebGPUToVulkanPassCase>{
|
|
// Decompose Initialized Variables
|
|
{// input
|
|
"OpCapability Shader\n"
|
|
"OpCapability VulkanMemoryModelKHR\n"
|
|
"OpExtension \"SPV_KHR_vulkan_memory_model\"\n"
|
|
"OpMemoryModel Logical VulkanKHR\n"
|
|
"OpEntryPoint Vertex %1 \"shader\"\n"
|
|
"%uint = OpTypeInt 32 0\n"
|
|
"%_ptr_Function_uint = OpTypePointer Function %uint\n"
|
|
"%4 = OpConstantNull %uint\n"
|
|
"%void = OpTypeVoid\n"
|
|
"%6 = OpTypeFunction %void\n"
|
|
"%1 = OpFunction %void None %6\n"
|
|
"%7 = OpLabel\n"
|
|
"%8 = OpVariable %_ptr_Function_uint Function %4\n"
|
|
"OpReturn\n"
|
|
"OpFunctionEnd\n",
|
|
// expected
|
|
"OpCapability Shader\n"
|
|
"OpCapability VulkanMemoryModelKHR\n"
|
|
"OpExtension \"SPV_KHR_vulkan_memory_model\"\n"
|
|
"OpMemoryModel Logical VulkanKHR\n"
|
|
"OpEntryPoint Vertex %1 \"shader\"\n"
|
|
"%uint = OpTypeInt 32 0\n"
|
|
"%_ptr_Function_uint = OpTypePointer Function %uint\n"
|
|
"%4 = OpConstantNull %uint\n"
|
|
"%void = OpTypeVoid\n"
|
|
"%6 = OpTypeFunction %void\n"
|
|
"%1 = OpFunction %void None %6\n"
|
|
"%7 = OpLabel\n"
|
|
"%8 = OpVariable %_ptr_Function_uint Function\n"
|
|
"OpStore %8 %4\n"
|
|
"OpReturn\n"
|
|
"OpFunctionEnd\n",
|
|
// pass
|
|
"decompose-initialized-variables"},
|
|
// Compact IDs
|
|
{// input
|
|
"OpCapability Shader\n"
|
|
"OpCapability VulkanMemoryModelKHR\n"
|
|
"OpExtension \"SPV_KHR_vulkan_memory_model\"\n"
|
|
"OpMemoryModel Logical VulkanKHR\n"
|
|
"OpEntryPoint Vertex %1000 \"shader\"\n"
|
|
"%10 = OpTypeVoid\n"
|
|
"%100 = OpTypeFunction %10\n"
|
|
"%1000 = OpFunction %10 None %100\n"
|
|
"%10000 = OpLabel\n"
|
|
"OpReturn\n"
|
|
"OpFunctionEnd\n",
|
|
// expected
|
|
"OpCapability Shader\n"
|
|
"OpCapability VulkanMemoryModelKHR\n"
|
|
"OpExtension \"SPV_KHR_vulkan_memory_model\"\n"
|
|
"OpMemoryModel Logical VulkanKHR\n"
|
|
"OpEntryPoint Vertex %1 \"shader\"\n"
|
|
"%void = OpTypeVoid\n"
|
|
"%3 = OpTypeFunction %void\n"
|
|
"%1 = OpFunction %void None %3\n"
|
|
"%4 = OpLabel\n"
|
|
"OpReturn\n"
|
|
"OpFunctionEnd\n",
|
|
// pass
|
|
"compact-ids"}}));
|
|
|
|
} // namespace
|
|
} // namespace opt
|
|
} // namespace spvtools
|