mirror of
https://github.com/KhronosGroup/SPIRV-Tools
synced 2024-11-24 12:30:13 +00:00
5dcb576b69
Motivated by integrating spirv-reduce into spirv-fuzz, so that an added function can be made smaller during shrinking, this adds support in spirv-reduce for asking reduction to be restricted to the instructions of a single specified function.
309 lines
10 KiB
C++
309 lines
10 KiB
C++
// 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_const_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(OperandToConstantReductionPassTest, BasicCheck) {
|
|
std::string prologue = R"(
|
|
OpCapability Shader
|
|
%1 = OpExtInstImport "GLSL.std.450"
|
|
OpMemoryModel Logical GLSL450
|
|
OpEntryPoint Fragment %4 "main" %37
|
|
OpExecutionMode %4 OriginUpperLeft
|
|
OpSource ESSL 310
|
|
OpName %4 "main"
|
|
OpName %9 "buf1"
|
|
OpMemberName %9 0 "f"
|
|
OpName %11 ""
|
|
OpName %24 "buf2"
|
|
OpMemberName %24 0 "i"
|
|
OpName %26 ""
|
|
OpName %37 "_GLF_color"
|
|
OpMemberDecorate %9 0 Offset 0
|
|
OpDecorate %9 Block
|
|
OpDecorate %11 DescriptorSet 0
|
|
OpDecorate %11 Binding 1
|
|
OpMemberDecorate %24 0 Offset 0
|
|
OpDecorate %24 Block
|
|
OpDecorate %26 DescriptorSet 0
|
|
OpDecorate %26 Binding 2
|
|
OpDecorate %37 Location 0
|
|
%2 = OpTypeVoid
|
|
%3 = OpTypeFunction %2
|
|
%6 = OpTypeFloat 32
|
|
%9 = OpTypeStruct %6
|
|
%10 = OpTypePointer Uniform %9
|
|
%11 = OpVariable %10 Uniform
|
|
%12 = OpTypeInt 32 1
|
|
%13 = OpConstant %12 0
|
|
%14 = OpTypePointer Uniform %6
|
|
%20 = OpConstant %6 2
|
|
%24 = OpTypeStruct %12
|
|
%25 = OpTypePointer Uniform %24
|
|
%26 = OpVariable %25 Uniform
|
|
%27 = OpTypePointer Uniform %12
|
|
%33 = OpConstant %12 3
|
|
%35 = OpTypeVector %6 4
|
|
%36 = OpTypePointer Output %35
|
|
%37 = OpVariable %36 Output
|
|
%4 = OpFunction %2 None %3
|
|
%5 = OpLabel
|
|
%15 = OpAccessChain %14 %11 %13
|
|
%16 = OpLoad %6 %15
|
|
%19 = OpFAdd %6 %16 %16
|
|
%21 = OpFAdd %6 %19 %20
|
|
%28 = OpAccessChain %27 %26 %13
|
|
%29 = OpLoad %12 %28
|
|
)";
|
|
|
|
std::string epilogue = R"(
|
|
%45 = OpConvertSToF %6 %34
|
|
%46 = OpCompositeConstruct %35 %16 %21 %43 %45
|
|
OpStore %37 %46
|
|
OpReturn
|
|
OpFunctionEnd
|
|
)";
|
|
|
|
std::string original = prologue + R"(
|
|
%32 = OpIAdd %12 %29 %29
|
|
%34 = OpIAdd %12 %32 %33
|
|
%43 = OpConvertSToF %6 %29
|
|
)" + epilogue;
|
|
|
|
std::string expected = prologue + R"(
|
|
%32 = OpIAdd %12 %13 %13 ; %29 -> %13 x 2
|
|
%34 = OpIAdd %12 %13 %33 ; %32 -> %13
|
|
%43 = OpConvertSToF %6 %13 ; %29 -> %13
|
|
)" + epilogue;
|
|
|
|
const auto env = SPV_ENV_UNIVERSAL_1_3;
|
|
const auto consumer = nullptr;
|
|
const auto context =
|
|
BuildModule(env, consumer, original, kReduceAssembleOption);
|
|
const auto ops =
|
|
OperandToConstReductionOpportunityFinder().GetAvailableOpportunities(
|
|
context.get(), 0);
|
|
ASSERT_EQ(17, ops.size());
|
|
ASSERT_TRUE(ops[0]->PreconditionHolds());
|
|
ops[0]->TryToApply();
|
|
ASSERT_TRUE(ops[1]->PreconditionHolds());
|
|
ops[1]->TryToApply();
|
|
ASSERT_TRUE(ops[2]->PreconditionHolds());
|
|
ops[2]->TryToApply();
|
|
ASSERT_TRUE(ops[3]->PreconditionHolds());
|
|
ops[3]->TryToApply();
|
|
|
|
CheckEqual(env, expected, context.get());
|
|
}
|
|
|
|
TEST(OperandToConstantReductionPassTest, WithCalledFunction) {
|
|
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
|
|
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 =
|
|
OperandToConstReductionOpportunityFinder().GetAvailableOpportunities(
|
|
context.get(), 0);
|
|
ASSERT_EQ(0, ops.size());
|
|
}
|
|
|
|
TEST(OperandToConstantReductionPassTest, TargetSpecificFunction) {
|
|
std::string shader = R"(
|
|
OpCapability Shader
|
|
%1 = OpExtInstImport "GLSL.std.450"
|
|
OpMemoryModel Logical GLSL450
|
|
OpEntryPoint Fragment %4 "main"
|
|
OpExecutionMode %4 OriginUpperLeft
|
|
OpSource ESSL 320
|
|
%2 = OpTypeVoid
|
|
%3 = OpTypeFunction %2
|
|
%6 = OpTypeInt 32 1
|
|
%7 = OpTypePointer Function %6
|
|
%8 = OpTypeFunction %6 %7
|
|
%17 = OpConstant %6 1
|
|
%20 = OpConstant %6 2
|
|
%23 = OpConstant %6 0
|
|
%24 = OpTypeBool
|
|
%35 = OpConstant %6 3
|
|
%53 = OpConstant %6 10
|
|
%4 = OpFunction %2 None %3
|
|
%5 = OpLabel
|
|
%65 = OpVariable %7 Function
|
|
%68 = OpVariable %7 Function
|
|
%73 = OpVariable %7 Function
|
|
OpStore %65 %35
|
|
%66 = OpLoad %6 %65
|
|
%67 = OpIAdd %6 %66 %17
|
|
OpStore %65 %67
|
|
%69 = OpLoad %6 %65
|
|
OpStore %68 %69
|
|
%70 = OpFunctionCall %6 %13 %68
|
|
%71 = OpLoad %6 %65
|
|
%72 = OpIAdd %6 %71 %70
|
|
OpStore %65 %72
|
|
%74 = OpLoad %6 %65
|
|
OpStore %73 %74
|
|
%75 = OpFunctionCall %6 %10 %73
|
|
%76 = OpLoad %6 %65
|
|
%77 = OpIAdd %6 %76 %75
|
|
OpStore %65 %77
|
|
OpReturn
|
|
OpFunctionEnd
|
|
%10 = OpFunction %6 None %8
|
|
%9 = OpFunctionParameter %7
|
|
%11 = OpLabel
|
|
%15 = OpVariable %7 Function
|
|
%16 = OpLoad %6 %9
|
|
%18 = OpIAdd %6 %16 %17
|
|
OpStore %15 %18
|
|
%19 = OpLoad %6 %15
|
|
%21 = OpIAdd %6 %19 %20
|
|
OpStore %15 %21
|
|
%22 = OpLoad %6 %15
|
|
%25 = OpSGreaterThan %24 %22 %23
|
|
OpSelectionMerge %27 None
|
|
OpBranchConditional %25 %26 %27
|
|
%26 = OpLabel
|
|
%28 = OpLoad %6 %9
|
|
OpReturnValue %28
|
|
%27 = OpLabel
|
|
%30 = OpLoad %6 %9
|
|
%31 = OpIAdd %6 %30 %17
|
|
OpReturnValue %31
|
|
OpFunctionEnd
|
|
%13 = OpFunction %6 None %8
|
|
%12 = OpFunctionParameter %7
|
|
%14 = OpLabel
|
|
%41 = OpVariable %7 Function
|
|
%46 = OpVariable %7 Function
|
|
%55 = OpVariable %7 Function
|
|
%34 = OpLoad %6 %12
|
|
%36 = OpIEqual %24 %34 %35
|
|
OpSelectionMerge %38 None
|
|
OpBranchConditional %36 %37 %38
|
|
%37 = OpLabel
|
|
%39 = OpLoad %6 %12
|
|
%40 = OpIMul %6 %20 %39
|
|
OpStore %41 %40
|
|
%42 = OpFunctionCall %6 %10 %41
|
|
OpReturnValue %42
|
|
%38 = OpLabel
|
|
%44 = OpLoad %6 %12
|
|
%45 = OpIAdd %6 %44 %17
|
|
OpStore %12 %45
|
|
OpStore %46 %23
|
|
OpBranch %47
|
|
%47 = OpLabel
|
|
OpLoopMerge %49 %50 None
|
|
OpBranch %51
|
|
%51 = OpLabel
|
|
%52 = OpLoad %6 %46
|
|
%54 = OpSLessThan %24 %52 %53
|
|
OpBranchConditional %54 %48 %49
|
|
%48 = OpLabel
|
|
%56 = OpLoad %6 %12
|
|
OpStore %55 %56
|
|
%57 = OpFunctionCall %6 %10 %55
|
|
%58 = OpLoad %6 %12
|
|
%59 = OpIAdd %6 %58 %57
|
|
OpStore %12 %59
|
|
OpBranch %50
|
|
%50 = OpLabel
|
|
%60 = OpLoad %6 %46
|
|
%61 = OpIAdd %6 %60 %17
|
|
OpStore %46 %61
|
|
OpBranch %47
|
|
%49 = OpLabel
|
|
%62 = OpLoad %6 %12
|
|
OpReturnValue %62
|
|
OpFunctionEnd
|
|
)";
|
|
|
|
const auto env = SPV_ENV_UNIVERSAL_1_3;
|
|
const auto consumer = nullptr;
|
|
const auto context =
|
|
BuildModule(env, consumer, shader, kReduceAssembleOption);
|
|
|
|
// Targeting all functions, there are quite a few opportunities. To avoid
|
|
// making the test too sensitive, we check that there are more than a number
|
|
// somewhat lower than the real number.
|
|
const auto all_ops =
|
|
OperandToConstReductionOpportunityFinder().GetAvailableOpportunities(
|
|
context.get(), 0);
|
|
ASSERT_TRUE(all_ops.size() > 100);
|
|
|
|
// Targeting individual functions, there are fewer opportunities. Again, we
|
|
// avoid checking against an exact number so that the test is not too
|
|
// sensitive.
|
|
const auto ops_for_function_4 =
|
|
OperandToConstReductionOpportunityFinder().GetAvailableOpportunities(
|
|
context.get(), 4);
|
|
const auto ops_for_function_10 =
|
|
OperandToConstReductionOpportunityFinder().GetAvailableOpportunities(
|
|
context.get(), 10);
|
|
const auto ops_for_function_13 =
|
|
OperandToConstReductionOpportunityFinder().GetAvailableOpportunities(
|
|
context.get(), 13);
|
|
ASSERT_TRUE(ops_for_function_4.size() < 60);
|
|
ASSERT_TRUE(ops_for_function_10.size() < 50);
|
|
ASSERT_TRUE(ops_for_function_13.size() < 80);
|
|
|
|
// The total number of opportunities should be the sum of the per-function
|
|
// opportunities.
|
|
ASSERT_EQ(all_ops.size(), ops_for_function_4.size() +
|
|
ops_for_function_10.size() +
|
|
ops_for_function_13.size());
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace reduce
|
|
} // namespace spvtools
|