// 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 #include #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 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 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 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 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 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 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 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 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_VULKAN_1_1); opt.RegisterVulkanToWebGPUPasses(); std::vector pass_names = opt.GetPassNames(); std::vector registered_passes; for (auto name = pass_names.begin(); name != pass_names.end(); ++name) registered_passes.push_back(*name); std::vector 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>; TEST_P(VulkanToWebGPUPassTest, Ran) { std::vector binary; { SpirvTools tools(SPV_ENV_VULKAN_1_1); tools.Assemble(GetParam().input, &binary); } Optimizer opt(SPV_ENV_VULKAN_1_1); opt.RegisterVulkanToWebGPUPasses(); std::vector optimized; class ValidatorOptions validator_options; ASSERT_TRUE(opt.Run(binary.data(), binary.size(), &optimized, validator_options, true)) << GetParam().input << "\n"; 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{ // FlattenDecorations {// input "OpCapability Shader\n" "OpCapability VulkanMemoryModel\n" "OpExtension \"SPV_KHR_vulkan_memory_model\"\n" "OpMemoryModel Logical Vulkan\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 VulkanMemoryModel\n" "OpExtension \"SPV_KHR_vulkan_memory_model\"\n" "OpMemoryModel Logical Vulkan\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 VulkanMemoryModel\n" "OpExtension \"SPV_KHR_vulkan_memory_model\"\n" "OpMemoryModel Logical Vulkan\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 VulkanMemoryModel\n" "OpExtension \"SPV_KHR_vulkan_memory_model\"\n" "OpMemoryModel Logical Vulkan\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 VulkanMemoryModel\n" "OpExtension \"SPV_KHR_vulkan_memory_model\"\n" "OpMemoryModel Logical Vulkan\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 VulkanMemoryModel\n" "OpExtension \"SPV_KHR_vulkan_memory_model\"\n" "OpMemoryModel Logical Vulkan\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 VulkanMemoryModel\n" "OpExtension \"SPV_KHR_vulkan_memory_model\"\n" "OpMemoryModel Logical Vulkan\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" " 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 VulkanMemoryModel\n" "OpExtension \"SPV_KHR_vulkan_memory_model\"\n" "OpMemoryModel Logical Vulkan\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 VulkanMemoryModel\n" "OpExtension \"SPV_KHR_vulkan_memory_model\"\n" "OpMemoryModel Logical Vulkan\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 VulkanMemoryModel\n" "OpExtension \"SPV_KHR_vulkan_memory_model\"\n" "OpMemoryModel Logical Vulkan\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 VulkanMemoryModel\n" "OpExtension \"SPV_KHR_vulkan_memory_model\"\n" "OpMemoryModel Logical Vulkan\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 VulkanMemoryModel\n" "OpExtension \"SPV_KHR_vulkan_memory_model\"\n" "OpMemoryModel Logical Vulkan\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 VulkanMemoryModel\n" "OpExtension \"SPV_KHR_vulkan_memory_model\"\n" "OpMemoryModel Logical Vulkan\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 VulkanMemoryModel\n" "OpExtension \"SPV_KHR_vulkan_memory_model\"\n" "OpMemoryModel Logical Vulkan\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 VulkanMemoryModel\n" "OpExtension \"SPV_KHR_vulkan_memory_model\"\n" "OpMemoryModel Logical Vulkan\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 VulkanMemoryModel\n" "OpExtension \"SPV_KHR_vulkan_memory_model\"\n" "OpMemoryModel Logical Vulkan\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_WEBGPU_0); opt.RegisterWebGPUToVulkanPasses(); std::vector pass_names = opt.GetPassNames(); std::vector registered_passes; for (auto name = pass_names.begin(); name != pass_names.end(); ++name) registered_passes.push_back(*name); std::vector 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>; TEST_P(WebGPUToVulkanPassTest, Ran) { std::vector binary; { SpirvTools tools(SPV_ENV_WEBGPU_0); tools.Assemble(GetParam().input, &binary); } Optimizer opt(SPV_ENV_WEBGPU_0); opt.RegisterWebGPUToVulkanPasses(); std::vector 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{ // Decompose Initialized Variables {// input "OpCapability Shader\n" "OpCapability VulkanMemoryModel\n" "OpExtension \"SPV_KHR_vulkan_memory_model\"\n" "OpMemoryModel Logical Vulkan\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 VulkanMemoryModel\n" "OpExtension \"SPV_KHR_vulkan_memory_model\"\n" "OpMemoryModel Logical Vulkan\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 VulkanMemoryModel\n" "OpExtension \"SPV_KHR_vulkan_memory_model\"\n" "OpMemoryModel Logical Vulkan\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 VulkanMemoryModel\n" "OpExtension \"SPV_KHR_vulkan_memory_model\"\n" "OpMemoryModel Logical Vulkan\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