| // Copyright (c) 2018 Google LLC. | |
| // | |
| // 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 "effcee/effcee.h" | |
| #include "gmock/gmock.h" | |
| #include "test/opt/pass_fixture.h" | |
| namespace spvtools { | |
| namespace opt { | |
| namespace { | |
| using UnswitchTest = PassTest<::testing::Test>; | |
| /* | |
| Generated from the following GLSL + --eliminate-local-multi-store | |
| #version 450 core | |
| uniform vec4 c; | |
| void main() { | |
| int i = 0; | |
| int j = 0; | |
| bool cond = c[0] == 0; | |
| for (; i < 10; i++, j++) { | |
| if (cond) { | |
| i++; | |
| } | |
| else { | |
| j++; | |
| } | |
| } | |
| } | |
| */ | |
| TEST_F(UnswitchTest, SimpleUnswitch) { | |
| const std::string text = R"( | |
| ; CHECK: [[cst_cond:%\w+]] = OpFOrdEqual | |
| ; CHECK-NEXT: OpSelectionMerge [[if_merge:%\w+]] None | |
| ; CHECK-NEXT: OpBranchConditional [[cst_cond]] [[loop_t:%\w+]] [[loop_f:%\w+]] | |
| ; Loop specialized for false. | |
| ; CHECK: [[loop_f]] = OpLabel | |
| ; CHECK-NEXT: OpBranch [[loop:%\w+]] | |
| ; CHECK: [[loop]] = OpLabel | |
| ; CHECK-NEXT: [[phi_i:%\w+]] = OpPhi %int %int_0 [[loop_f]] [[iv_i:%\w+]] [[continue:%\w+]] | |
| ; CHECK-NEXT: [[phi_j:%\w+]] = OpPhi %int %int_0 [[loop_f]] [[iv_j:%\w+]] [[continue]] | |
| ; CHECK-NEXT: OpLoopMerge [[merge:%\w+]] [[continue]] None | |
| ; CHECK: [[loop_exit:%\w+]] = OpSLessThan {{%\w+}} [[phi_i]] {{%\w+}} | |
| ; CHECK-NEXT: OpBranchConditional [[loop_exit]] [[loop_body:%\w+]] [[merge]] | |
| ; [[loop_body]] = OpLabel | |
| ; CHECK: OpSelectionMerge [[sel_merge:%\w+]] None | |
| ; CHECK: OpBranchConditional %false [[bb1:%\w+]] [[bb2:%\w+]] | |
| ; CHECK: [[bb2]] = OpLabel | |
| ; CHECK-NEXT: [[inc_j:%\w+]] = OpIAdd %int [[phi_j]] %int_1 | |
| ; CHECK-NEXT: OpBranch [[sel_merge]] | |
| ; CHECK: [[bb1]] = OpLabel | |
| ; CHECK-NEXT: [[inc_i:%\w+]] = OpIAdd %int [[phi_i]] %int_1 | |
| ; CHECK-NEXT: OpBranch [[sel_merge]] | |
| ; CHECK: [[sel_merge]] = OpLabel | |
| ; CHECK: OpBranch [[if_merge]] | |
| ; Loop specialized for true. | |
| ; CHECK: [[loop_t]] = OpLabel | |
| ; CHECK-NEXT: OpBranch [[loop:%\w+]] | |
| ; CHECK: [[loop]] = OpLabel | |
| ; CHECK-NEXT: [[phi_i:%\w+]] = OpPhi %int %int_0 [[loop_t]] [[iv_i:%\w+]] [[continue:%\w+]] | |
| ; CHECK-NEXT: [[phi_j:%\w+]] = OpPhi %int %int_0 [[loop_t]] [[iv_j:%\w+]] [[continue]] | |
| ; CHECK-NEXT: OpLoopMerge [[merge:%\w+]] [[continue]] None | |
| ; CHECK: [[loop_exit:%\w+]] = OpSLessThan {{%\w+}} [[phi_i]] {{%\w+}} | |
| ; CHECK-NEXT: OpBranchConditional [[loop_exit]] [[loop_body:%\w+]] [[merge]] | |
| ; [[loop_body]] = OpLabel | |
| ; CHECK: OpSelectionMerge [[sel_merge:%\w+]] None | |
| ; CHECK: OpBranchConditional %true [[bb1:%\w+]] [[bb2:%\w+]] | |
| ; CHECK: [[bb1]] = OpLabel | |
| ; CHECK-NEXT: [[inc_i:%\w+]] = OpIAdd %int [[phi_i]] %int_1 | |
| ; CHECK-NEXT: OpBranch [[sel_merge]] | |
| ; CHECK: [[bb2]] = OpLabel | |
| ; CHECK-NEXT: [[inc_j:%\w+]] = OpIAdd %int [[phi_j]] %int_1 | |
| ; CHECK-NEXT: OpBranch [[sel_merge]] | |
| ; CHECK: [[sel_merge]] = OpLabel | |
| ; CHECK: OpBranch [[if_merge]] | |
| ; CHECK: [[if_merge]] = OpLabel | |
| ; CHECK-NEXT: OpReturn | |
| OpCapability Shader | |
| %1 = OpExtInstImport "GLSL.std.450" | |
| OpMemoryModel Logical GLSL450 | |
| OpEntryPoint Fragment %main "main" | |
| OpExecutionMode %main OriginLowerLeft | |
| OpSource GLSL 450 | |
| OpName %main "main" | |
| OpName %c "c" | |
| OpDecorate %c Location 0 | |
| OpDecorate %c DescriptorSet 0 | |
| %void = OpTypeVoid | |
| %3 = OpTypeFunction %void | |
| %int = OpTypeInt 32 1 | |
| %_ptr_Function_int = OpTypePointer Function %int | |
| %int_0 = OpConstant %int 0 | |
| %bool = OpTypeBool | |
| %_ptr_Function_bool = OpTypePointer Function %bool | |
| %float = OpTypeFloat 32 | |
| %v4float = OpTypeVector %float 4 | |
| %_ptr_UniformConstant_v4float = OpTypePointer UniformConstant %v4float | |
| %c = OpVariable %_ptr_UniformConstant_v4float UniformConstant | |
| %uint = OpTypeInt 32 0 | |
| %uint_0 = OpConstant %uint 0 | |
| %_ptr_UniformConstant_float = OpTypePointer UniformConstant %float | |
| %float_0 = OpConstant %float 0 | |
| %int_10 = OpConstant %int 10 | |
| %int_1 = OpConstant %int 1 | |
| %main = OpFunction %void None %3 | |
| %5 = OpLabel | |
| %21 = OpAccessChain %_ptr_UniformConstant_float %c %uint_0 | |
| %22 = OpLoad %float %21 | |
| %24 = OpFOrdEqual %bool %22 %float_0 | |
| OpBranch %25 | |
| %25 = OpLabel | |
| %46 = OpPhi %int %int_0 %5 %43 %28 | |
| %47 = OpPhi %int %int_0 %5 %45 %28 | |
| OpLoopMerge %27 %28 None | |
| OpBranch %29 | |
| %29 = OpLabel | |
| %32 = OpSLessThan %bool %46 %int_10 | |
| OpBranchConditional %32 %26 %27 | |
| %26 = OpLabel | |
| OpSelectionMerge %35 None | |
| OpBranchConditional %24 %34 %39 | |
| %34 = OpLabel | |
| %38 = OpIAdd %int %46 %int_1 | |
| OpBranch %35 | |
| %39 = OpLabel | |
| %41 = OpIAdd %int %47 %int_1 | |
| OpBranch %35 | |
| %35 = OpLabel | |
| %48 = OpPhi %int %38 %34 %46 %39 | |
| %49 = OpPhi %int %47 %34 %41 %39 | |
| OpBranch %28 | |
| %28 = OpLabel | |
| %43 = OpIAdd %int %48 %int_1 | |
| %45 = OpIAdd %int %49 %int_1 | |
| OpBranch %25 | |
| %27 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| )"; | |
| SinglePassRunAndMatch<LoopUnswitchPass>(text, true); | |
| } | |
| /* | |
| Generated from the following GLSL + --eliminate-local-multi-store | |
| #version 330 core | |
| in vec4 c; | |
| void main() { | |
| int i = 0; | |
| bool cond = c[0] == 0; | |
| for (; i < 10; i++) { | |
| if (cond) { | |
| i++; | |
| } | |
| else { | |
| return; | |
| } | |
| } | |
| } | |
| */ | |
| TEST_F(UnswitchTest, UnswitchExit) { | |
| const std::string text = R"( | |
| ; CHECK: [[cst_cond:%\w+]] = OpFOrdEqual | |
| ; CHECK-NEXT: OpSelectionMerge [[if_merge:%\w+]] None | |
| ; CHECK-NEXT: OpBranchConditional [[cst_cond]] [[loop_t:%\w+]] [[loop_f:%\w+]] | |
| ; Loop specialized for false. | |
| ; CHECK: [[loop_f]] = OpLabel | |
| ; CHECK: OpReturn | |
| ; Loop specialized for true. | |
| ; CHECK: [[loop_t]] = OpLabel | |
| ; CHECK-NEXT: OpBranch [[loop:%\w+]] | |
| ; CHECK: [[loop]] = OpLabel | |
| ; CHECK-NEXT: [[phi_i:%\w+]] = OpPhi %int %int_0 [[loop_t]] [[iv_i:%\w+]] [[continue:%\w+]] | |
| ; CHECK-NEXT: OpLoopMerge [[merge:%\w+]] [[continue]] None | |
| ; CHECK: [[loop_exit:%\w+]] = OpSLessThan {{%\w+}} [[phi_i]] {{%\w+}} | |
| ; CHECK-NEXT: OpBranchConditional [[loop_exit]] {{%\w+}} [[merge]] | |
| ; Check that we have i+=2. | |
| ; CHECK: [[phi_i:%\w+]] = OpIAdd %int [[phi_i]] %int_1 | |
| ; CHECK: [[iv_i]] = OpIAdd %int [[phi_i]] %int_1 | |
| ; CHECK: [[merge]] = OpLabel | |
| ; CHECK-NEXT: OpBranch [[if_merge]] | |
| ; CHECK: [[if_merge]] = OpLabel | |
| ; CHECK-NEXT: OpReturn | |
| OpCapability Shader | |
| %1 = OpExtInstImport "GLSL.std.450" | |
| OpMemoryModel Logical GLSL450 | |
| OpEntryPoint Fragment %main "main" %c | |
| OpExecutionMode %main OriginUpperLeft | |
| OpSource GLSL 330 | |
| OpName %main "main" | |
| OpName %c "c" | |
| OpDecorate %c Location 0 | |
| OpDecorate %23 Uniform | |
| %void = OpTypeVoid | |
| %3 = OpTypeFunction %void | |
| %int = OpTypeInt 32 1 | |
| %_ptr_Function_int = OpTypePointer Function %int | |
| %int_0 = OpConstant %int 0 | |
| %bool = OpTypeBool | |
| %_ptr_Function_bool = OpTypePointer Function %bool | |
| %float = OpTypeFloat 32 | |
| %v4float = OpTypeVector %float 4 | |
| %_ptr_Input_v4float = OpTypePointer Input %v4float | |
| %c = OpVariable %_ptr_Input_v4float Input | |
| %uint = OpTypeInt 32 0 | |
| %uint_0 = OpConstant %uint 0 | |
| %_ptr_Input_float = OpTypePointer Input %float | |
| %float_0 = OpConstant %float 0 | |
| %int_10 = OpConstant %int 10 | |
| %int_1 = OpConstant %int 1 | |
| %main = OpFunction %void None %3 | |
| %5 = OpLabel | |
| %20 = OpAccessChain %_ptr_Input_float %c %uint_0 | |
| %21 = OpLoad %float %20 | |
| %23 = OpFOrdEqual %bool %21 %float_0 | |
| OpBranch %24 | |
| %24 = OpLabel | |
| %42 = OpPhi %int %int_0 %5 %41 %27 | |
| OpLoopMerge %26 %27 None | |
| OpBranch %28 | |
| %28 = OpLabel | |
| %31 = OpSLessThan %bool %42 %int_10 | |
| OpBranchConditional %31 %25 %26 | |
| %25 = OpLabel | |
| OpSelectionMerge %34 None | |
| OpBranchConditional %23 %33 %38 | |
| %33 = OpLabel | |
| %37 = OpIAdd %int %42 %int_1 | |
| OpBranch %34 | |
| %38 = OpLabel | |
| OpReturn | |
| %34 = OpLabel | |
| OpBranch %27 | |
| %27 = OpLabel | |
| %41 = OpIAdd %int %37 %int_1 | |
| OpBranch %24 | |
| %26 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| )"; | |
| SinglePassRunAndMatch<LoopUnswitchPass>(text, true); | |
| } | |
| /* | |
| Generated from the following GLSL + --eliminate-local-multi-store | |
| #version 330 core | |
| in vec4 c; | |
| void main() { | |
| int i = 0; | |
| bool cond = c[0] == 0; | |
| for (; i < 10; i++) { | |
| if (cond) { | |
| continue; | |
| } | |
| else { | |
| i++; | |
| } | |
| } | |
| } | |
| */ | |
| TEST_F(UnswitchTest, UnswitchContinue) { | |
| const std::string text = R"( | |
| ; CHECK: [[cst_cond:%\w+]] = OpFOrdEqual | |
| ; CHECK-NEXT: OpSelectionMerge [[if_merge:%\w+]] None | |
| ; CHECK-NEXT: OpBranchConditional [[cst_cond]] [[loop_t:%\w+]] [[loop_f:%\w+]] | |
| ; Loop specialized for false. | |
| ; CHECK: [[loop_f]] = OpLabel | |
| ; CHECK-NEXT: OpBranch [[loop:%\w+]] | |
| ; CHECK: [[loop]] = OpLabel | |
| ; CHECK-NEXT: [[phi_i:%\w+]] = OpPhi %int %int_0 [[loop_f]] [[iv_i:%\w+]] [[continue:%\w+]] | |
| ; CHECK-NEXT: OpLoopMerge [[merge:%\w+]] [[continue]] None | |
| ; CHECK: [[loop_exit:%\w+]] = OpSLessThan {{%\w+}} [[phi_i]] {{%\w+}} | |
| ; CHECK-NEXT: OpBranchConditional [[loop_exit]] [[loop_body:%\w+]] [[merge]] | |
| ; CHECK: [[loop_body:%\w+]] = OpLabel | |
| ; CHECK-NEXT: OpSelectionMerge | |
| ; CHECK-NEXT: OpBranchConditional %false | |
| ; CHECK: [[merge]] = OpLabel | |
| ; CHECK-NEXT: OpBranch [[if_merge]] | |
| ; Loop specialized for true. | |
| ; CHECK: [[loop_t]] = OpLabel | |
| ; CHECK-NEXT: OpBranch [[loop:%\w+]] | |
| ; CHECK: [[loop]] = OpLabel | |
| ; CHECK-NEXT: [[phi_i:%\w+]] = OpPhi %int %int_0 [[loop_t]] [[iv_i:%\w+]] [[continue:%\w+]] | |
| ; CHECK-NEXT: OpLoopMerge [[merge:%\w+]] [[continue]] None | |
| ; CHECK: [[loop_exit:%\w+]] = OpSLessThan {{%\w+}} [[phi_i]] {{%\w+}} | |
| ; CHECK-NEXT: OpBranchConditional [[loop_exit]] [[loop_body:%\w+]] [[merge]] | |
| ; CHECK: [[loop_body:%\w+]] = OpLabel | |
| ; CHECK-NEXT: OpSelectionMerge | |
| ; CHECK-NEXT: OpBranchConditional %true | |
| ; CHECK: [[merge]] = OpLabel | |
| ; CHECK-NEXT: OpBranch [[if_merge]] | |
| ; CHECK: [[if_merge]] = OpLabel | |
| ; CHECK-NEXT: OpReturn | |
| OpCapability Shader | |
| %1 = OpExtInstImport "GLSL.std.450" | |
| OpMemoryModel Logical GLSL450 | |
| OpEntryPoint Fragment %main "main" %c | |
| OpExecutionMode %main OriginUpperLeft | |
| OpSource GLSL 330 | |
| OpName %main "main" | |
| OpName %c "c" | |
| OpDecorate %c Location 0 | |
| OpDecorate %23 Uniform | |
| %void = OpTypeVoid | |
| %3 = OpTypeFunction %void | |
| %int = OpTypeInt 32 1 | |
| %_ptr_Function_int = OpTypePointer Function %int | |
| %int_0 = OpConstant %int 0 | |
| %bool = OpTypeBool | |
| %_ptr_Function_bool = OpTypePointer Function %bool | |
| %float = OpTypeFloat 32 | |
| %v4float = OpTypeVector %float 4 | |
| %_ptr_Input_v4float = OpTypePointer Input %v4float | |
| %c = OpVariable %_ptr_Input_v4float Input | |
| %uint = OpTypeInt 32 0 | |
| %uint_0 = OpConstant %uint 0 | |
| %_ptr_Input_float = OpTypePointer Input %float | |
| %float_0 = OpConstant %float 0 | |
| %int_10 = OpConstant %int 10 | |
| %int_1 = OpConstant %int 1 | |
| %main = OpFunction %void None %3 | |
| %5 = OpLabel | |
| %20 = OpAccessChain %_ptr_Input_float %c %uint_0 | |
| %21 = OpLoad %float %20 | |
| %23 = OpFOrdEqual %bool %21 %float_0 | |
| OpBranch %24 | |
| %24 = OpLabel | |
| %42 = OpPhi %int %int_0 %5 %41 %27 | |
| OpLoopMerge %26 %27 None | |
| OpBranch %28 | |
| %28 = OpLabel | |
| %31 = OpSLessThan %bool %42 %int_10 | |
| OpBranchConditional %31 %25 %26 | |
| %25 = OpLabel | |
| OpSelectionMerge %34 None | |
| OpBranchConditional %23 %33 %36 | |
| %33 = OpLabel | |
| OpBranch %27 | |
| %36 = OpLabel | |
| %39 = OpIAdd %int %42 %int_1 | |
| OpBranch %34 | |
| %34 = OpLabel | |
| OpBranch %27 | |
| %27 = OpLabel | |
| %43 = OpPhi %int %42 %33 %39 %34 | |
| %41 = OpIAdd %int %43 %int_1 | |
| OpBranch %24 | |
| %26 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| )"; | |
| SinglePassRunAndMatch<LoopUnswitchPass>(text, true); | |
| } | |
| /* | |
| Generated from the following GLSL + --eliminate-local-multi-store | |
| #version 330 core | |
| in vec4 c; | |
| void main() { | |
| int i = 0; | |
| bool cond = c[0] == 0; | |
| for (; i < 10; i++) { | |
| if (cond) { | |
| i++; | |
| } | |
| else { | |
| break; | |
| } | |
| } | |
| } | |
| */ | |
| TEST_F(UnswitchTest, UnswitchKillLoop) { | |
| const std::string text = R"( | |
| ; CHECK: [[cst_cond:%\w+]] = OpFOrdEqual | |
| ; CHECK-NEXT: OpSelectionMerge [[if_merge:%\w+]] None | |
| ; CHECK-NEXT: OpBranchConditional [[cst_cond]] [[loop_t:%\w+]] [[loop_f:%\w+]] | |
| ; Loop specialized for false. | |
| ; CHECK: [[loop_f]] = OpLabel | |
| ; CHECK: OpBranch [[if_merge]] | |
| ; Loop specialized for true. | |
| ; CHECK: [[loop_t]] = OpLabel | |
| ; CHECK-NEXT: OpBranch [[loop:%\w+]] | |
| ; CHECK: [[loop]] = OpLabel | |
| ; CHECK-NEXT: [[phi_i:%\w+]] = OpPhi %int %int_0 [[loop_t]] [[iv_i:%\w+]] [[continue:%\w+]] | |
| ; CHECK-NEXT: OpLoopMerge [[merge:%\w+]] [[continue]] None | |
| ; CHECK: [[loop_exit:%\w+]] = OpSLessThan {{%\w+}} [[phi_i]] {{%\w+}} | |
| ; CHECK-NEXT: OpBranchConditional [[loop_exit]] {{%\w+}} [[merge]] | |
| ; Check that we have i+=2. | |
| ; CHECK: [[phi_i:%\w+]] = OpIAdd %int [[phi_i]] %int_1 | |
| ; CHECK: [[iv_i]] = OpIAdd %int [[phi_i]] %int_1 | |
| ; CHECK: [[merge]] = OpLabel | |
| ; CHECK-NEXT: OpBranch [[if_merge]] | |
| ; CHECK: [[if_merge]] = OpLabel | |
| ; CHECK-NEXT: OpReturn | |
| OpCapability Shader | |
| %1 = OpExtInstImport "GLSL.std.450" | |
| OpMemoryModel Logical GLSL450 | |
| OpEntryPoint Fragment %main "main" %c | |
| OpExecutionMode %main OriginUpperLeft | |
| OpSource GLSL 330 | |
| OpName %main "main" | |
| OpName %c "c" | |
| OpDecorate %c Location 0 | |
| OpDecorate %23 Uniform | |
| %void = OpTypeVoid | |
| %3 = OpTypeFunction %void | |
| %int = OpTypeInt 32 1 | |
| %_ptr_Function_int = OpTypePointer Function %int | |
| %int_0 = OpConstant %int 0 | |
| %bool = OpTypeBool | |
| %_ptr_Function_bool = OpTypePointer Function %bool | |
| %float = OpTypeFloat 32 | |
| %v4float = OpTypeVector %float 4 | |
| %_ptr_Input_v4float = OpTypePointer Input %v4float | |
| %c = OpVariable %_ptr_Input_v4float Input | |
| %uint = OpTypeInt 32 0 | |
| %uint_0 = OpConstant %uint 0 | |
| %_ptr_Input_float = OpTypePointer Input %float | |
| %float_0 = OpConstant %float 0 | |
| %int_10 = OpConstant %int 10 | |
| %int_1 = OpConstant %int 1 | |
| %main = OpFunction %void None %3 | |
| %5 = OpLabel | |
| %20 = OpAccessChain %_ptr_Input_float %c %uint_0 | |
| %21 = OpLoad %float %20 | |
| %23 = OpFOrdEqual %bool %21 %float_0 | |
| OpBranch %24 | |
| %24 = OpLabel | |
| %42 = OpPhi %int %int_0 %5 %41 %27 | |
| OpLoopMerge %26 %27 None | |
| OpBranch %28 | |
| %28 = OpLabel | |
| %31 = OpSLessThan %bool %42 %int_10 | |
| OpBranchConditional %31 %25 %26 | |
| %25 = OpLabel | |
| OpSelectionMerge %34 None | |
| OpBranchConditional %23 %33 %38 | |
| %33 = OpLabel | |
| %37 = OpIAdd %int %42 %int_1 | |
| OpBranch %34 | |
| %38 = OpLabel | |
| OpBranch %26 | |
| %34 = OpLabel | |
| OpBranch %27 | |
| %27 = OpLabel | |
| %41 = OpIAdd %int %37 %int_1 | |
| OpBranch %24 | |
| %26 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| )"; | |
| SinglePassRunAndMatch<LoopUnswitchPass>(text, true); | |
| } | |
| /* | |
| Generated from the following GLSL + --eliminate-local-multi-store | |
| #version 330 core | |
| in vec4 c; | |
| void main() { | |
| int i = 0; | |
| int cond = int(c[0]); | |
| for (; i < 10; i++) { | |
| switch (cond) { | |
| case 0: | |
| return; | |
| case 1: | |
| discard; | |
| case 2: | |
| break; | |
| default: | |
| break; | |
| } | |
| } | |
| bool cond2 = i == 9; | |
| } | |
| */ | |
| TEST_F(UnswitchTest, UnswitchSwitch) { | |
| const std::string text = R"( | |
| ; CHECK: [[cst_cond:%\w+]] = OpConvertFToS | |
| ; CHECK-NEXT: OpSelectionMerge [[if_merge:%\w+]] None | |
| ; CHECK-NEXT: OpSwitch [[cst_cond]] [[default:%\w+]] 0 [[loop_0:%\w+]] 1 [[loop_1:%\w+]] 2 [[loop_2:%\w+]] | |
| ; Loop specialized for 2. | |
| ; CHECK: [[loop_2]] = OpLabel | |
| ; CHECK-NEXT: OpBranch [[loop:%\w+]] | |
| ; CHECK: [[loop]] = OpLabel | |
| ; CHECK-NEXT: [[phi_i:%\w+]] = OpPhi %int %int_0 [[loop_2]] [[iv_i:%\w+]] [[continue:%\w+]] | |
| ; CHECK-NEXT: OpLoopMerge [[merge:%\w+]] [[continue]] None | |
| ; CHECK: [[loop_exit:%\w+]] = OpSLessThan {{%\w+}} [[phi_i]] {{%\w+}} | |
| ; CHECK-NEXT: OpBranchConditional [[loop_exit]] [[loop_body:%\w+]] [[merge]] | |
| ; CHECK: [[loop_body]] = OpLabel | |
| ; CHECK-NEXT: OpSelectionMerge | |
| ; CHECK-NEXT: OpSwitch %int_2 | |
| ; CHECK: [[merge]] = OpLabel | |
| ; CHECK-NEXT: OpBranch [[if_merge]] | |
| ; Loop specialized for 1. | |
| ; CHECK: [[loop_1]] = OpLabel | |
| ; CHECK-NEXT: OpBranch [[loop:%\w+]] | |
| ; CHECK: [[loop]] = OpLabel | |
| ; CHECK-NEXT: [[phi_i:%\w+]] = OpPhi %int %int_0 [[loop_1]] [[iv_i:%\w+]] [[continue:%\w+]] | |
| ; CHECK-NEXT: OpLoopMerge [[merge:%\w+]] [[continue]] None | |
| ; CHECK: [[loop_exit:%\w+]] = OpSLessThan {{%\w+}} [[phi_i]] {{%\w+}} | |
| ; CHECK-NEXT: OpBranchConditional [[loop_exit]] [[loop_body:%\w+]] [[merge]] | |
| ; CHECK: [[loop_body]] = OpLabel | |
| ; CHECK-NEXT: OpSelectionMerge | |
| ; CHECK-NEXT: OpSwitch %int_1 | |
| ; CHECK: [[merge]] = OpLabel | |
| ; CHECK-NEXT: OpBranch [[if_merge]] | |
| ; Loop specialized for 0. | |
| ; CHECK: [[loop_0]] = OpLabel | |
| ; CHECK-NEXT: OpBranch [[loop:%\w+]] | |
| ; CHECK: [[loop]] = OpLabel | |
| ; CHECK-NEXT: [[phi_i:%\w+]] = OpPhi %int %int_0 [[loop_0]] [[iv_i:%\w+]] [[continue:%\w+]] | |
| ; CHECK-NEXT: OpLoopMerge [[merge:%\w+]] [[continue]] None | |
| ; CHECK: [[loop_exit:%\w+]] = OpSLessThan {{%\w+}} [[phi_i]] {{%\w+}} | |
| ; CHECK-NEXT: OpBranchConditional [[loop_exit]] [[loop_body:%\w+]] [[merge]] | |
| ; CHECK: [[loop_body]] = OpLabel | |
| ; CHECK-NEXT: OpSelectionMerge | |
| ; CHECK-NEXT: OpSwitch %int_0 | |
| ; CHECK: [[merge]] = OpLabel | |
| ; CHECK-NEXT: OpBranch [[if_merge]] | |
| ; Loop specialized for the default case. | |
| ; CHECK: [[default]] = OpLabel | |
| ; CHECK-NEXT: OpBranch [[loop:%\w+]] | |
| ; CHECK: [[loop]] = OpLabel | |
| ; CHECK-NEXT: [[phi_i:%\w+]] = OpPhi %int %int_0 [[default]] [[iv_i:%\w+]] [[continue:%\w+]] | |
| ; CHECK-NEXT: OpLoopMerge [[merge:%\w+]] [[continue]] None | |
| ; CHECK: [[loop_exit:%\w+]] = OpSLessThan {{%\w+}} [[phi_i]] {{%\w+}} | |
| ; CHECK-NEXT: OpBranchConditional [[loop_exit]] [[loop_body:%\w+]] [[merge]] | |
| ; CHECK: [[loop_body]] = OpLabel | |
| ; CHECK-NEXT: OpSelectionMerge | |
| ; CHECK-NEXT: OpSwitch %uint_3 | |
| ; CHECK: [[merge]] = OpLabel | |
| ; CHECK-NEXT: OpBranch [[if_merge]] | |
| ; CHECK: [[if_merge]] = OpLabel | |
| ; CHECK-NEXT: OpReturn | |
| OpCapability Shader | |
| %1 = OpExtInstImport "GLSL.std.450" | |
| OpMemoryModel Logical GLSL450 | |
| OpEntryPoint Fragment %main "main" %c | |
| OpExecutionMode %main OriginUpperLeft | |
| OpSource GLSL 330 | |
| OpName %main "main" | |
| OpName %c "c" | |
| OpDecorate %c Location 0 | |
| OpDecorate %20 Uniform | |
| %void = OpTypeVoid | |
| %3 = OpTypeFunction %void | |
| %int = OpTypeInt 32 1 | |
| %_ptr_Function_int = OpTypePointer Function %int | |
| %int_0 = OpConstant %int 0 | |
| %float = OpTypeFloat 32 | |
| %v4float = OpTypeVector %float 4 | |
| %_ptr_Input_v4float = OpTypePointer Input %v4float | |
| %c = OpVariable %_ptr_Input_v4float Input | |
| %uint = OpTypeInt 32 0 | |
| %uint_0 = OpConstant %uint 0 | |
| %_ptr_Input_float = OpTypePointer Input %float | |
| %int_10 = OpConstant %int 10 | |
| %bool = OpTypeBool | |
| %int_1 = OpConstant %int 1 | |
| %_ptr_Function_bool = OpTypePointer Function %bool | |
| %main = OpFunction %void None %3 | |
| %5 = OpLabel | |
| %18 = OpAccessChain %_ptr_Input_float %c %uint_0 | |
| %19 = OpLoad %float %18 | |
| %20 = OpConvertFToS %int %19 | |
| OpBranch %21 | |
| %21 = OpLabel | |
| %49 = OpPhi %int %int_0 %5 %43 %24 | |
| OpLoopMerge %23 %24 None | |
| OpBranch %25 | |
| %25 = OpLabel | |
| %29 = OpSLessThan %bool %49 %int_10 | |
| OpBranchConditional %29 %22 %23 | |
| %22 = OpLabel | |
| OpSelectionMerge %35 None | |
| OpSwitch %20 %34 0 %31 1 %32 2 %33 | |
| %34 = OpLabel | |
| OpBranch %35 | |
| %31 = OpLabel | |
| OpReturn | |
| %32 = OpLabel | |
| OpKill | |
| %33 = OpLabel | |
| OpBranch %35 | |
| %35 = OpLabel | |
| OpBranch %24 | |
| %24 = OpLabel | |
| %43 = OpIAdd %int %49 %int_1 | |
| OpBranch %21 | |
| %23 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| )"; | |
| SinglePassRunAndMatch<LoopUnswitchPass>(text, true); | |
| } | |
| /* | |
| Generated from the following GLSL + --eliminate-local-multi-store | |
| #version 440 core | |
| layout(location = 0)in vec4 c; | |
| void main() { | |
| int i = 0; | |
| int j = 0; | |
| int k = 0; | |
| bool cond = c[0] == 0; | |
| for (; i < 10; i++) { | |
| for (; j < 10; j++) { | |
| if (cond) { | |
| i++; | |
| } else { | |
| j++; | |
| } | |
| } | |
| } | |
| } | |
| */ | |
| TEST_F(UnswitchTest, UnSwitchNested) { | |
| // Test that an branch can be unswitched out of two nested loops. | |
| const std::string text = R"( | |
| ; CHECK: [[cst_cond:%\w+]] = OpFOrdEqual | |
| ; CHECK-NEXT: OpSelectionMerge [[if_merge:%\w+]] None | |
| ; CHECK-NEXT: OpBranchConditional [[cst_cond]] [[loop_t:%\w+]] [[loop_f:%\w+]] | |
| ; Loop specialized for false | |
| ; CHECK: [[loop_f]] = OpLabel | |
| ; CHECK-NEXT: OpBranch [[loop:%\w+]] | |
| ; CHECK: [[loop]] = OpLabel | |
| ; CHECK-NEXT: {{%\w+}} = OpPhi %int %int_0 [[loop_f]] {{%\w+}} [[continue:%\w+]] | |
| ; CHECK-NEXT: {{%\w+}} = OpPhi %int %int_0 [[loop_f]] {{%\w+}} [[continue]] | |
| ; CHECK-NEXT: OpLoopMerge [[merge:%\w+]] [[continue]] None | |
| ; CHECK-NOT: [[merge]] = OpLabel | |
| ; CHECK: OpLoopMerge | |
| ; CHECK-NEXT: OpBranch [[bb1:%\w+]] | |
| ; CHECK: [[bb1]] = OpLabel | |
| ; CHECK-NEXT: OpSLessThan | |
| ; CHECK-NEXT: OpBranchConditional {{%\w+}} [[bb2:%\w+]] | |
| ; CHECK: [[bb2]] = OpLabel | |
| ; CHECK-NEXT: OpSelectionMerge | |
| ; CHECK-NEXT: OpBranchConditional %false | |
| ; CHECK: [[merge]] = OpLabel | |
| ; Loop specialized for true. Same as first loop except the branch condition is true. | |
| ; CHECK: [[loop_t]] = OpLabel | |
| ; CHECK-NEXT: OpBranch [[loop:%\w+]] | |
| ; CHECK: [[loop]] = OpLabel | |
| ; CHECK-NEXT: {{%\w+}} = OpPhi %int %int_0 [[loop_t]] {{%\w+}} [[continue:%\w+]] | |
| ; CHECK-NEXT: {{%\w+}} = OpPhi %int %int_0 [[loop_t]] {{%\w+}} [[continue]] | |
| ; CHECK-NEXT: OpLoopMerge [[merge:%\w+]] [[continue]] None | |
| ; CHECK-NOT: [[merge]] = OpLabel | |
| ; CHECK: OpLoopMerge | |
| ; CHECK-NEXT: OpBranch [[bb1:%\w+]] | |
| ; CHECK: [[bb1]] = OpLabel | |
| ; CHECK-NEXT: OpSLessThan | |
| ; CHECK-NEXT: OpBranchConditional {{%\w+}} [[bb2:%\w+]] | |
| ; CHECK: [[bb2]] = OpLabel | |
| ; CHECK-NEXT: OpSelectionMerge | |
| ; CHECK-NEXT: OpBranchConditional %true | |
| ; CHECK: [[merge]] = OpLabel | |
| OpCapability Shader | |
| %1 = OpExtInstImport "GLSL.std.450" | |
| OpMemoryModel Logical GLSL450 | |
| OpEntryPoint Fragment %main "main" %c | |
| OpExecutionMode %main OriginUpperLeft | |
| OpSource GLSL 440 | |
| OpName %main "main" | |
| OpName %c "c" | |
| OpDecorate %c Location 0 | |
| OpDecorate %25 Uniform | |
| %void = OpTypeVoid | |
| %3 = OpTypeFunction %void | |
| %int = OpTypeInt 32 1 | |
| %_ptr_Function_int = OpTypePointer Function %int | |
| %int_0 = OpConstant %int 0 | |
| %bool = OpTypeBool | |
| %_ptr_Function_bool = OpTypePointer Function %bool | |
| %float = OpTypeFloat 32 | |
| %v4float = OpTypeVector %float 4 | |
| %_ptr_Input_v4float = OpTypePointer Input %v4float | |
| %c = OpVariable %_ptr_Input_v4float Input | |
| %uint = OpTypeInt 32 0 | |
| %uint_0 = OpConstant %uint 0 | |
| %_ptr_Input_float = OpTypePointer Input %float | |
| %float_0 = OpConstant %float 0 | |
| %int_10 = OpConstant %int 10 | |
| %int_1 = OpConstant %int 1 | |
| %main = OpFunction %void None %3 | |
| %5 = OpLabel | |
| %22 = OpAccessChain %_ptr_Input_float %c %uint_0 | |
| %23 = OpLoad %float %22 | |
| %25 = OpFOrdEqual %bool %23 %float_0 | |
| OpBranch %26 | |
| %26 = OpLabel | |
| %67 = OpPhi %int %int_0 %5 %52 %29 | |
| %68 = OpPhi %int %int_0 %5 %70 %29 | |
| OpLoopMerge %28 %29 None | |
| OpBranch %30 | |
| %30 = OpLabel | |
| %33 = OpSLessThan %bool %67 %int_10 | |
| OpBranchConditional %33 %27 %28 | |
| %27 = OpLabel | |
| OpBranch %34 | |
| %34 = OpLabel | |
| %69 = OpPhi %int %67 %27 %46 %37 | |
| %70 = OpPhi %int %68 %27 %50 %37 | |
| OpLoopMerge %36 %37 None | |
| OpBranch %38 | |
| %38 = OpLabel | |
| %40 = OpSLessThan %bool %70 %int_10 | |
| OpBranchConditional %40 %35 %36 | |
| %35 = OpLabel | |
| OpSelectionMerge %43 None | |
| OpBranchConditional %25 %42 %47 | |
| %42 = OpLabel | |
| %46 = OpIAdd %int %69 %int_1 | |
| OpBranch %43 | |
| %47 = OpLabel | |
| OpReturn | |
| %43 = OpLabel | |
| OpBranch %37 | |
| %37 = OpLabel | |
| %50 = OpIAdd %int %70 %int_1 | |
| OpBranch %34 | |
| %36 = OpLabel | |
| OpBranch %29 | |
| %29 = OpLabel | |
| %52 = OpIAdd %int %69 %int_1 | |
| OpBranch %26 | |
| %28 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| )"; | |
| SinglePassRunAndMatch<LoopUnswitchPass>(text, true); | |
| } | |
| /* | |
| Generated from the following GLSL + --eliminate-local-multi-store | |
| #version 330 core | |
| in vec4 c; | |
| void main() { | |
| bool cond = false; | |
| if (c[0] == 0) { | |
| cond = c[1] == 0; | |
| } else { | |
| cond = c[2] == 0; | |
| } | |
| for (int i = 0; i < 10; i++) { | |
| if (cond) { | |
| i++; | |
| } | |
| } | |
| } | |
| */ | |
| TEST_F(UnswitchTest, UnswitchNotUniform) { | |
| // Check that the unswitch is not triggered (condition loop invariant but not | |
| // uniform) | |
| const std::string text = R"( | |
| OpCapability Shader | |
| %1 = OpExtInstImport "GLSL.std.450" | |
| OpMemoryModel Logical GLSL450 | |
| OpEntryPoint Fragment %main "main" %c | |
| OpExecutionMode %main OriginUpperLeft | |
| OpSource GLSL 330 | |
| OpName %main "main" | |
| OpName %c "c" | |
| OpDecorate %c Location 0 | |
| %void = OpTypeVoid | |
| %3 = OpTypeFunction %void | |
| %bool = OpTypeBool | |
| %_ptr_Function_bool = OpTypePointer Function %bool | |
| %float = OpTypeFloat 32 | |
| %v4float = OpTypeVector %float 4 | |
| %_ptr_Input_v4float = OpTypePointer Input %v4float | |
| %c = OpVariable %_ptr_Input_v4float Input | |
| %uint = OpTypeInt 32 0 | |
| %uint_0 = OpConstant %uint 0 | |
| %_ptr_Input_float = OpTypePointer Input %float | |
| %float_0 = OpConstant %float 0 | |
| %uint_1 = OpConstant %uint 1 | |
| %uint_2 = OpConstant %uint 2 | |
| %int = OpTypeInt 32 1 | |
| %_ptr_Function_int = OpTypePointer Function %int | |
| %int_0 = OpConstant %int 0 | |
| %int_10 = OpConstant %int 10 | |
| %int_1 = OpConstant %int 1 | |
| %main = OpFunction %void None %3 | |
| %5 = OpLabel | |
| %17 = OpAccessChain %_ptr_Input_float %c %uint_0 | |
| %18 = OpLoad %float %17 | |
| %20 = OpFOrdEqual %bool %18 %float_0 | |
| OpSelectionMerge %22 None | |
| OpBranchConditional %20 %21 %27 | |
| %21 = OpLabel | |
| %24 = OpAccessChain %_ptr_Input_float %c %uint_1 | |
| %25 = OpLoad %float %24 | |
| %26 = OpFOrdEqual %bool %25 %float_0 | |
| OpBranch %22 | |
| %27 = OpLabel | |
| %29 = OpAccessChain %_ptr_Input_float %c %uint_2 | |
| %30 = OpLoad %float %29 | |
| %31 = OpFOrdEqual %bool %30 %float_0 | |
| OpBranch %22 | |
| %22 = OpLabel | |
| %52 = OpPhi %bool %26 %21 %31 %27 | |
| OpBranch %36 | |
| %36 = OpLabel | |
| %53 = OpPhi %int %int_0 %22 %51 %39 | |
| OpLoopMerge %38 %39 None | |
| OpBranch %40 | |
| %40 = OpLabel | |
| %43 = OpSLessThan %bool %53 %int_10 | |
| OpBranchConditional %43 %37 %38 | |
| %37 = OpLabel | |
| OpSelectionMerge %46 None | |
| OpBranchConditional %52 %45 %46 | |
| %45 = OpLabel | |
| %49 = OpIAdd %int %53 %int_1 | |
| OpBranch %46 | |
| %46 = OpLabel | |
| %54 = OpPhi %int %53 %37 %49 %45 | |
| OpBranch %39 | |
| %39 = OpLabel | |
| %51 = OpIAdd %int %54 %int_1 | |
| OpBranch %36 | |
| %38 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| )"; | |
| auto result = | |
| SinglePassRunAndDisassemble<LoopUnswitchPass>(text, true, false); | |
| EXPECT_EQ(Pass::Status::SuccessWithoutChange, std::get<1>(result)); | |
| } | |
| TEST_F(UnswitchTest, DontUnswitchLatch) { | |
| // Check that the unswitch is not triggered for the latch branch. | |
| const std::string text = R"( | |
| OpCapability Shader | |
| %1 = OpExtInstImport "GLSL.std.450" | |
| OpMemoryModel Logical GLSL450 | |
| OpEntryPoint Fragment %4 "main" | |
| OpExecutionMode %4 OriginUpperLeft | |
| OpSource ESSL 310 | |
| %void = OpTypeVoid | |
| %3 = OpTypeFunction %void | |
| %bool = OpTypeBool | |
| %false = OpConstantFalse %bool | |
| %4 = OpFunction %void None %3 | |
| %5 = OpLabel | |
| OpBranch %6 | |
| %6 = OpLabel | |
| OpLoopMerge %8 %9 None | |
| OpBranch %7 | |
| %7 = OpLabel | |
| OpBranch %9 | |
| %9 = OpLabel | |
| OpBranchConditional %false %6 %8 | |
| %8 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| )"; | |
| auto result = | |
| SinglePassRunAndDisassemble<LoopUnswitchPass>(text, true, false); | |
| EXPECT_EQ(Pass::Status::SuccessWithoutChange, std::get<1>(result)); | |
| } | |
| TEST_F(UnswitchTest, DontUnswitchConstantCondition) { | |
| const std::string text = R"( | |
| OpCapability Shader | |
| %1 = OpExtInstImport "GLSL.std.450" | |
| OpMemoryModel Logical GLSL450 | |
| OpEntryPoint Fragment %main "main" | |
| OpExecutionMode %main OriginLowerLeft | |
| OpSource GLSL 450 | |
| OpName %main "main" | |
| %void = OpTypeVoid | |
| %4 = OpTypeFunction %void | |
| %int = OpTypeInt 32 1 | |
| %int_0 = OpConstant %int 0 | |
| %bool = OpTypeBool | |
| %true = OpConstantTrue %bool | |
| %int_1 = OpConstant %int 1 | |
| %main = OpFunction %void None %4 | |
| %10 = OpLabel | |
| OpBranch %11 | |
| %11 = OpLabel | |
| %12 = OpPhi %int %int_0 %10 %13 %14 | |
| OpLoopMerge %15 %14 None | |
| OpBranch %16 | |
| %16 = OpLabel | |
| %17 = OpSLessThan %bool %12 %int_1 | |
| OpBranchConditional %17 %18 %15 | |
| %18 = OpLabel | |
| OpSelectionMerge %19 None | |
| OpBranchConditional %true %20 %19 | |
| %20 = OpLabel | |
| %21 = OpIAdd %int %12 %int_1 | |
| OpBranch %19 | |
| %19 = OpLabel | |
| %22 = OpPhi %int %21 %20 %12 %18 | |
| OpBranch %14 | |
| %14 = OpLabel | |
| %13 = OpIAdd %int %22 %int_1 | |
| OpBranch %11 | |
| %15 = OpLabel | |
| OpReturn | |
| OpFunctionEnd | |
| )"; | |
| auto result = | |
| SinglePassRunAndDisassemble<LoopUnswitchPass>(text, true, false); | |
| EXPECT_EQ(Pass::Status::SuccessWithoutChange, std::get<1>(result)); | |
| } | |
| } // namespace | |
| } // namespace opt | |
| } // namespace spvtools |