| // 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 "source/reduce/operand_to_undef_reduction_opportunity_finder.h" |
| |
| #include "source/opt/build_module.h" |
| #include "source/reduce/reduction_opportunity.h" |
| #include "test/reduce/reduce_test_util.h" |
| |
| namespace spvtools { |
| namespace reduce { |
| namespace { |
| |
| TEST(OperandToUndefReductionPassTest, BasicCheck) { |
| // The following shader has 10 opportunities for replacing with undef. |
| |
| // #version 310 es |
| // |
| // precision highp float; |
| // |
| // layout(location=0) out vec4 _GLF_color; |
| // |
| // layout(set = 0, binding = 0) uniform buf0 { |
| // vec2 uniform1; |
| // }; |
| // |
| // void main() |
| // { |
| // _GLF_color = |
| // vec4( // opportunity |
| // uniform1.x / 2.0, // opportunity x2 (2.0 is const) |
| // uniform1.y / uniform1.x, // opportunity x3 |
| // uniform1.x + uniform1.x, // opportunity x3 |
| // uniform1.y); // opportunity |
| // } |
| |
| std::string original = R"( |
| OpCapability Shader |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %4 "main" %9 |
| OpExecutionMode %4 OriginUpperLeft |
| OpSource ESSL 310 |
| OpName %4 "main" |
| OpName %9 "_GLF_color" |
| OpName %11 "buf0" |
| OpMemberName %11 0 "uniform1" |
| OpName %13 "" |
| OpDecorate %9 Location 0 |
| OpMemberDecorate %11 0 Offset 0 |
| OpDecorate %11 Block |
| OpDecorate %13 DescriptorSet 0 |
| OpDecorate %13 Binding 0 |
| %2 = OpTypeVoid |
| %3 = OpTypeFunction %2 |
| %6 = OpTypeFloat 32 |
| %7 = OpTypeVector %6 4 |
| %8 = OpTypePointer Output %7 |
| %9 = OpVariable %8 Output |
| %10 = OpTypeVector %6 2 |
| %11 = OpTypeStruct %10 |
| %12 = OpTypePointer Uniform %11 |
| %13 = OpVariable %12 Uniform |
| %14 = OpTypeInt 32 1 |
| %15 = OpConstant %14 0 |
| %16 = OpTypeInt 32 0 |
| %17 = OpConstant %16 0 |
| %18 = OpTypePointer Uniform %6 |
| %21 = OpConstant %6 2 |
| %23 = OpConstant %16 1 |
| %4 = OpFunction %2 None %3 |
| %5 = OpLabel |
| %19 = OpAccessChain %18 %13 %15 %17 |
| %20 = OpLoad %6 %19 |
| %22 = OpFDiv %6 %20 %21 ; opportunity %20 (%21 is const) |
| %24 = OpAccessChain %18 %13 %15 %23 |
| %25 = OpLoad %6 %24 |
| %26 = OpAccessChain %18 %13 %15 %17 |
| %27 = OpLoad %6 %26 |
| %28 = OpFDiv %6 %25 %27 ; opportunity %25 %27 |
| %29 = OpAccessChain %18 %13 %15 %17 |
| %30 = OpLoad %6 %29 |
| %31 = OpAccessChain %18 %13 %15 %17 |
| %32 = OpLoad %6 %31 |
| %33 = OpFAdd %6 %30 %32 ; opportunity %30 %32 |
| %34 = OpAccessChain %18 %13 %15 %23 |
| %35 = OpLoad %6 %34 |
| %36 = OpCompositeConstruct %7 %22 %28 %33 %35 ; opportunity %22 %28 %33 %35 |
| OpStore %9 %36 ; opportunity %36 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| // This is the same as original, except where noted. |
| std::string expected = R"( |
| OpCapability Shader |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %4 "main" %9 |
| OpExecutionMode %4 OriginUpperLeft |
| OpSource ESSL 310 |
| OpName %4 "main" |
| OpName %9 "_GLF_color" |
| OpName %11 "buf0" |
| OpMemberName %11 0 "uniform1" |
| OpName %13 "" |
| OpDecorate %9 Location 0 |
| OpMemberDecorate %11 0 Offset 0 |
| OpDecorate %11 Block |
| OpDecorate %13 DescriptorSet 0 |
| OpDecorate %13 Binding 0 |
| %2 = OpTypeVoid |
| %3 = OpTypeFunction %2 |
| %6 = OpTypeFloat 32 |
| %7 = OpTypeVector %6 4 |
| %8 = OpTypePointer Output %7 |
| %9 = OpVariable %8 Output |
| %10 = OpTypeVector %6 2 |
| %11 = OpTypeStruct %10 |
| %12 = OpTypePointer Uniform %11 |
| %13 = OpVariable %12 Uniform |
| %14 = OpTypeInt 32 1 |
| %15 = OpConstant %14 0 |
| %16 = OpTypeInt 32 0 |
| %17 = OpConstant %16 0 |
| %18 = OpTypePointer Uniform %6 |
| %21 = OpConstant %6 2 |
| %23 = OpConstant %16 1 |
| %37 = OpUndef %6 ; Added undef float as %37 |
| %4 = OpFunction %2 None %3 |
| %5 = OpLabel |
| %19 = OpAccessChain %18 %13 %15 %17 |
| %20 = OpLoad %6 %19 |
| %22 = OpFDiv %6 %37 %21 ; Replaced with %37 |
| %24 = OpAccessChain %18 %13 %15 %23 |
| %25 = OpLoad %6 %24 |
| %26 = OpAccessChain %18 %13 %15 %17 |
| %27 = OpLoad %6 %26 |
| %28 = OpFDiv %6 %37 %37 ; Replaced with %37 twice |
| %29 = OpAccessChain %18 %13 %15 %17 |
| %30 = OpLoad %6 %29 |
| %31 = OpAccessChain %18 %13 %15 %17 |
| %32 = OpLoad %6 %31 |
| %33 = OpFAdd %6 %30 %32 |
| %34 = OpAccessChain %18 %13 %15 %23 |
| %35 = OpLoad %6 %34 |
| %36 = OpCompositeConstruct %7 %22 %28 %33 %35 |
| OpStore %9 %36 |
| OpReturn |
| OpFunctionEnd |
| )"; |
| |
| const auto env = SPV_ENV_UNIVERSAL_1_3; |
| const auto consumer = nullptr; |
| const auto context = |
| BuildModule(env, consumer, original, kReduceAssembleOption); |
| const auto ops = |
| OperandToUndefReductionOpportunityFinder().GetAvailableOpportunities( |
| context.get(), 0); |
| |
| ASSERT_EQ(10, ops.size()); |
| |
| // Apply first three opportunities. |
| ASSERT_TRUE(ops[0]->PreconditionHolds()); |
| ops[0]->TryToApply(); |
| ASSERT_TRUE(ops[1]->PreconditionHolds()); |
| ops[1]->TryToApply(); |
| ASSERT_TRUE(ops[2]->PreconditionHolds()); |
| ops[2]->TryToApply(); |
| |
| CheckEqual(env, expected, context.get()); |
| } |
| |
| TEST(OperandToUndefReductionPassTest, WithCalledFunction) { |
| // The following shader has no opportunities. |
| // Most importantly, the noted function operand is not changed. |
| |
| std::string shader = R"( |
| OpCapability Shader |
| %1 = OpExtInstImport "GLSL.std.450" |
| OpMemoryModel Logical GLSL450 |
| OpEntryPoint Fragment %4 "main" %10 %12 |
| OpExecutionMode %4 OriginUpperLeft |
| OpSource ESSL 310 |
| %2 = OpTypeVoid |
| %3 = OpTypeFunction %2 |
| %6 = OpTypeFloat 32 |
| %7 = OpTypeVector %6 4 |
| %8 = OpTypeFunction %7 |
| %9 = OpTypePointer Output %7 |
| %10 = OpVariable %9 Output |
| %11 = OpTypePointer Input %7 |
| %12 = OpVariable %11 Input |
| %13 = OpConstant %6 0 |
| %14 = OpConstantComposite %7 %13 %13 %13 %13 |
| %4 = OpFunction %2 None %3 |
| %5 = OpLabel |
| %15 = OpFunctionCall %7 %16 ; do not replace %16 with undef |
| OpReturn |
| OpFunctionEnd |
| %16 = OpFunction %7 None %8 |
| %17 = OpLabel |
| OpReturnValue %14 |
| OpFunctionEnd |
| )"; |
| |
| const auto env = SPV_ENV_UNIVERSAL_1_3; |
| const auto consumer = nullptr; |
| const auto context = |
| BuildModule(env, consumer, shader, kReduceAssembleOption); |
| const auto ops = |
| OperandToUndefReductionOpportunityFinder().GetAvailableOpportunities( |
| context.get(), 0); |
| ASSERT_EQ(0, ops.size()); |
| } |
| |
| } // namespace |
| } // namespace reduce |
| } // namespace spvtools |