SPIRV-Tools/test/fuzz/transformation_outline_function_test.cpp
Alastair Donaldson 2e41d5ece7
spirv-fuzz: Avoid passing access chains as parameters (#3112)
This change prevents the spirv-fuzz function outliner from outlining a
region that uses the result of an OpAccessChain not defined inside the
region.  Such accesses were turning into parameters to the outlined
function, and the result of an OpAccessChain cannot be passed as a
function parameter according to the SPIR-V specification.
2019-12-19 17:34:36 +00:00

2046 lines
67 KiB
C++

// Copyright (c) 2019 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/fuzz/transformation_outline_function.h"
#include "test/fuzz/fuzz_test_util.h"
namespace spvtools {
namespace fuzz {
namespace {
TEST(TransformationOutlineFunctionTest, TrivialOutline) {
// This tests outlining of a single, empty basic block.
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%4 = OpFunction %2 None %3
%5 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_4;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(5, 5, /* not relevant */ 200,
100, 101, 102, 103,
/* not relevant */ 201, {}, {});
ASSERT_TRUE(transformation.IsApplicable(context.get(), fact_manager));
transformation.Apply(context.get(), &fact_manager);
ASSERT_TRUE(IsValid(env, context.get()));
std::string after_transformation = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%4 = OpFunction %2 None %3
%5 = OpLabel
%103 = OpFunctionCall %2 %101
OpReturn
OpFunctionEnd
%101 = OpFunction %2 None %3
%102 = OpLabel
OpReturn
OpFunctionEnd
)";
ASSERT_TRUE(IsEqual(env, after_transformation, context.get()));
}
TEST(TransformationOutlineFunctionTest,
DoNotOutlineIfRegionStartsWithOpVariable) {
// This checks that we do not outline the first block of a function if it
// contains OpVariable.
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%7 = OpTypeBool
%8 = OpTypePointer Function %7
%4 = OpFunction %2 None %3
%5 = OpLabel
%6 = OpVariable %8 Function
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_4;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(5, 5, /* not relevant */ 200,
100, 101, 102, 103,
/* not relevant */ 201, {}, {});
ASSERT_FALSE(transformation.IsApplicable(context.get(), fact_manager));
}
TEST(TransformationOutlineFunctionTest, OutlineInterestingControlFlowNoState) {
// This tests outlining of some non-trivial control flow, but such that the
// basic blocks in the control flow do not actually do anything.
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%20 = OpTypeBool
%21 = OpConstantTrue %20
%3 = OpTypeFunction %2
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %6
%6 = OpLabel
OpBranch %7
%7 = OpLabel
OpSelectionMerge %9 None
OpBranchConditional %21 %8 %9
%8 = OpLabel
OpBranch %9
%9 = OpLabel
OpLoopMerge %12 %11 None
OpBranch %10
%10 = OpLabel
OpBranchConditional %21 %11 %12
%11 = OpLabel
OpBranch %9
%12 = OpLabel
OpBranch %13
%13 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_4;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(6, 13, /* not relevant */
200, 100, 101, 102, 103,
/* not relevant */ 201, {}, {});
ASSERT_TRUE(transformation.IsApplicable(context.get(), fact_manager));
transformation.Apply(context.get(), &fact_manager);
ASSERT_TRUE(IsValid(env, context.get()));
std::string after_transformation = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%20 = OpTypeBool
%21 = OpConstantTrue %20
%3 = OpTypeFunction %2
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %6
%6 = OpLabel
%103 = OpFunctionCall %2 %101
OpReturn
OpFunctionEnd
%101 = OpFunction %2 None %3
%102 = OpLabel
OpBranch %7
%7 = OpLabel
OpSelectionMerge %9 None
OpBranchConditional %21 %8 %9
%8 = OpLabel
OpBranch %9
%9 = OpLabel
OpLoopMerge %12 %11 None
OpBranch %10
%10 = OpLabel
OpBranchConditional %21 %11 %12
%11 = OpLabel
OpBranch %9
%12 = OpLabel
OpBranch %13
%13 = OpLabel
OpReturn
OpFunctionEnd
)";
ASSERT_TRUE(IsEqual(env, after_transformation, context.get()));
}
TEST(TransformationOutlineFunctionTest, OutlineCodeThatGeneratesUnusedIds) {
// This tests outlining of a single basic block that does some computation,
// but that does not use nor generate ids required outside of the outlined
// region.
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%20 = OpTypeInt 32 1
%21 = OpConstant %20 5
%3 = OpTypeFunction %2
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %6
%6 = OpLabel
%7 = OpCopyObject %20 %21
%8 = OpCopyObject %20 %21
%9 = OpIAdd %20 %7 %8
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_4;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(6, 6, /* not relevant */ 200,
100, 101, 102, 103,
/* not relevant */ 201, {}, {});
ASSERT_TRUE(transformation.IsApplicable(context.get(), fact_manager));
transformation.Apply(context.get(), &fact_manager);
ASSERT_TRUE(IsValid(env, context.get()));
std::string after_transformation = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%20 = OpTypeInt 32 1
%21 = OpConstant %20 5
%3 = OpTypeFunction %2
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %6
%6 = OpLabel
%103 = OpFunctionCall %2 %101
OpReturn
OpFunctionEnd
%101 = OpFunction %2 None %3
%102 = OpLabel
%7 = OpCopyObject %20 %21
%8 = OpCopyObject %20 %21
%9 = OpIAdd %20 %7 %8
OpReturn
OpFunctionEnd
)";
ASSERT_TRUE(IsEqual(env, after_transformation, context.get()));
}
TEST(TransformationOutlineFunctionTest, OutlineCodeThatGeneratesSingleUsedId) {
// This tests outlining of a block that generates an id that is used in a
// later block.
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%20 = OpTypeInt 32 1
%21 = OpConstant %20 5
%3 = OpTypeFunction %2
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %6
%6 = OpLabel
%7 = OpCopyObject %20 %21
%8 = OpCopyObject %20 %21
%9 = OpIAdd %20 %7 %8
OpBranch %10
%10 = OpLabel
%11 = OpCopyObject %20 %9
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_4;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(6, 6, 99, 100, 101, 102, 103,
105, {}, {{9, 104}});
ASSERT_TRUE(transformation.IsApplicable(context.get(), fact_manager));
transformation.Apply(context.get(), &fact_manager);
ASSERT_TRUE(IsValid(env, context.get()));
std::string after_transformation = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%20 = OpTypeInt 32 1
%21 = OpConstant %20 5
%3 = OpTypeFunction %2
%99 = OpTypeStruct %20
%100 = OpTypeFunction %99
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %6
%6 = OpLabel
%103 = OpFunctionCall %99 %101
%9 = OpCompositeExtract %20 %103 0
OpBranch %10
%10 = OpLabel
%11 = OpCopyObject %20 %9
OpReturn
OpFunctionEnd
%101 = OpFunction %99 None %100
%102 = OpLabel
%7 = OpCopyObject %20 %21
%8 = OpCopyObject %20 %21
%104 = OpIAdd %20 %7 %8
%105 = OpCompositeConstruct %99 %104
OpReturnValue %105
OpFunctionEnd
)";
ASSERT_TRUE(IsEqual(env, after_transformation, context.get()));
}
TEST(TransformationOutlineFunctionTest, OutlineDiamondThatGeneratesSeveralIds) {
// This tests outlining of several blocks that generate a number of ids that
// are used in later blocks.
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%20 = OpTypeInt 32 1
%21 = OpConstant %20 5
%22 = OpTypeBool
%3 = OpTypeFunction %2
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %6
%6 = OpLabel
%7 = OpCopyObject %20 %21
%8 = OpCopyObject %20 %21
%9 = OpSLessThan %22 %7 %8
OpSelectionMerge %12 None
OpBranchConditional %9 %10 %11
%10 = OpLabel
%13 = OpIAdd %20 %7 %8
OpBranch %12
%11 = OpLabel
%14 = OpIAdd %20 %7 %7
OpBranch %12
%12 = OpLabel
%15 = OpPhi %20 %13 %10 %14 %11
OpBranch %80
%80 = OpLabel
OpBranch %16
%16 = OpLabel
%17 = OpCopyObject %20 %15
%18 = OpCopyObject %22 %9
%19 = OpIAdd %20 %7 %8
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_4;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(
6, 80, 100, 101, 102, 103, 104, 105, {},
{{15, 106}, {9, 107}, {7, 108}, {8, 109}});
ASSERT_TRUE(transformation.IsApplicable(context.get(), fact_manager));
transformation.Apply(context.get(), &fact_manager);
ASSERT_TRUE(IsValid(env, context.get()));
std::string after_transformation = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%20 = OpTypeInt 32 1
%21 = OpConstant %20 5
%22 = OpTypeBool
%3 = OpTypeFunction %2
%100 = OpTypeStruct %20 %20 %22 %20
%101 = OpTypeFunction %100
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %6
%6 = OpLabel
%104 = OpFunctionCall %100 %102
%7 = OpCompositeExtract %20 %104 0
%8 = OpCompositeExtract %20 %104 1
%9 = OpCompositeExtract %22 %104 2
%15 = OpCompositeExtract %20 %104 3
OpBranch %16
%16 = OpLabel
%17 = OpCopyObject %20 %15
%18 = OpCopyObject %22 %9
%19 = OpIAdd %20 %7 %8
OpReturn
OpFunctionEnd
%102 = OpFunction %100 None %101
%103 = OpLabel
%108 = OpCopyObject %20 %21
%109 = OpCopyObject %20 %21
%107 = OpSLessThan %22 %108 %109
OpSelectionMerge %12 None
OpBranchConditional %107 %10 %11
%10 = OpLabel
%13 = OpIAdd %20 %108 %109
OpBranch %12
%11 = OpLabel
%14 = OpIAdd %20 %108 %108
OpBranch %12
%12 = OpLabel
%106 = OpPhi %20 %13 %10 %14 %11
OpBranch %80
%80 = OpLabel
%105 = OpCompositeConstruct %100 %108 %109 %107 %106
OpReturnValue %105
OpFunctionEnd
)";
ASSERT_TRUE(IsEqual(env, after_transformation, context.get()));
}
TEST(TransformationOutlineFunctionTest, OutlineCodeThatUsesASingleId) {
// This tests outlining of a block that uses an id defined earlier.
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%20 = OpTypeInt 32 1
%21 = OpConstant %20 5
%3 = OpTypeFunction %2
%4 = OpFunction %2 None %3
%5 = OpLabel
%7 = OpCopyObject %20 %21
OpBranch %6
%6 = OpLabel
%8 = OpCopyObject %20 %7
OpBranch %10
%10 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_4;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(6, 6, 100, 101, 102, 103, 104,
105, {{7, 106}}, {});
ASSERT_TRUE(transformation.IsApplicable(context.get(), fact_manager));
transformation.Apply(context.get(), &fact_manager);
ASSERT_TRUE(IsValid(env, context.get()));
std::string after_transformation = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%20 = OpTypeInt 32 1
%21 = OpConstant %20 5
%3 = OpTypeFunction %2
%101 = OpTypeFunction %2 %20
%4 = OpFunction %2 None %3
%5 = OpLabel
%7 = OpCopyObject %20 %21
OpBranch %6
%6 = OpLabel
%104 = OpFunctionCall %2 %102 %7
OpBranch %10
%10 = OpLabel
OpReturn
OpFunctionEnd
%102 = OpFunction %2 None %101
%106 = OpFunctionParameter %20
%103 = OpLabel
%8 = OpCopyObject %20 %106
OpReturn
OpFunctionEnd
)";
ASSERT_TRUE(IsEqual(env, after_transformation, context.get()));
}
TEST(TransformationOutlineFunctionTest, OutlineCodeThatUsesAVariable) {
// This tests outlining of a block that uses a variable.
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%20 = OpTypeInt 32 1
%21 = OpConstant %20 5
%3 = OpTypeFunction %2
%12 = OpTypePointer Function %20
%4 = OpFunction %2 None %3
%5 = OpLabel
%13 = OpVariable %12 Function
OpBranch %6
%6 = OpLabel
%8 = OpLoad %20 %13
OpBranch %10
%10 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_4;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(6, 6, 100, 101, 102, 103, 104,
105, {{13, 106}}, {});
ASSERT_TRUE(transformation.IsApplicable(context.get(), fact_manager));
transformation.Apply(context.get(), &fact_manager);
ASSERT_TRUE(IsValid(env, context.get()));
std::string after_transformation = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%20 = OpTypeInt 32 1
%21 = OpConstant %20 5
%3 = OpTypeFunction %2
%12 = OpTypePointer Function %20
%101 = OpTypeFunction %2 %12
%4 = OpFunction %2 None %3
%5 = OpLabel
%13 = OpVariable %12 Function
OpBranch %6
%6 = OpLabel
%104 = OpFunctionCall %2 %102 %13
OpBranch %10
%10 = OpLabel
OpReturn
OpFunctionEnd
%102 = OpFunction %2 None %101
%106 = OpFunctionParameter %12
%103 = OpLabel
%8 = OpLoad %20 %106
OpReturn
OpFunctionEnd
)";
ASSERT_TRUE(IsEqual(env, after_transformation, context.get()));
}
TEST(TransformationOutlineFunctionTest, OutlineCodeThatUsesAParameter) {
// This tests outlining of a block that uses a function parameter.
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
OpName %10 "foo(i1;"
OpName %9 "x"
OpName %18 "param"
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%6 = OpTypeInt 32 1
%7 = OpTypePointer Function %6
%8 = OpTypeFunction %6 %7
%13 = OpConstant %6 1
%17 = OpConstant %6 3
%4 = OpFunction %2 None %3
%5 = OpLabel
%18 = OpVariable %7 Function
OpStore %18 %17
%19 = OpFunctionCall %6 %10 %18
OpReturn
OpFunctionEnd
%10 = OpFunction %6 None %8
%9 = OpFunctionParameter %7
%11 = OpLabel
%12 = OpLoad %6 %9
%14 = OpIAdd %6 %12 %13
OpReturnValue %14
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_4;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(11, 11, 100, 101, 102, 103, 104,
105, {{9, 106}}, {{14, 107}});
ASSERT_TRUE(transformation.IsApplicable(context.get(), fact_manager));
transformation.Apply(context.get(), &fact_manager);
ASSERT_TRUE(IsValid(env, context.get()));
std::string after_transformation = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
OpName %10 "foo(i1;"
OpName %9 "x"
OpName %18 "param"
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%6 = OpTypeInt 32 1
%7 = OpTypePointer Function %6
%8 = OpTypeFunction %6 %7
%13 = OpConstant %6 1
%17 = OpConstant %6 3
%100 = OpTypeStruct %6
%101 = OpTypeFunction %100 %7
%4 = OpFunction %2 None %3
%5 = OpLabel
%18 = OpVariable %7 Function
OpStore %18 %17
%19 = OpFunctionCall %6 %10 %18
OpReturn
OpFunctionEnd
%10 = OpFunction %6 None %8
%9 = OpFunctionParameter %7
%11 = OpLabel
%104 = OpFunctionCall %100 %102 %9
%14 = OpCompositeExtract %6 %104 0
OpReturnValue %14
OpFunctionEnd
%102 = OpFunction %100 None %101
%106 = OpFunctionParameter %7
%103 = OpLabel
%12 = OpLoad %6 %106
%107 = OpIAdd %6 %12 %13
%105 = OpCompositeConstruct %100 %107
OpReturnValue %105
OpFunctionEnd
)";
ASSERT_TRUE(IsEqual(env, after_transformation, context.get()));
}
TEST(TransformationOutlineFunctionTest,
DoNotOutlineIfLoopMergeIsOutsideRegion) {
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%9 = OpTypeBool
%10 = OpConstantTrue %9
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %6
%6 = OpLabel
OpLoopMerge %7 %8 None
OpBranch %8
%8 = OpLabel
OpBranchConditional %10 %6 %7
%7 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_4;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(6, 8, 100, 101, 102, 103, 104,
105, {}, {});
ASSERT_FALSE(transformation.IsApplicable(context.get(), fact_manager));
}
TEST(TransformationOutlineFunctionTest, DoNotOutlineIfRegionInvolvesReturn) {
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%20 = OpTypeBool
%21 = OpConstantTrue %20
%3 = OpTypeFunction %2
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %6
%6 = OpLabel
OpBranch %7
%7 = OpLabel
OpSelectionMerge %10 None
OpBranchConditional %21 %8 %9
%8 = OpLabel
OpReturn
%9 = OpLabel
OpBranch %10
%10 = OpLabel
OpBranch %11
%11 = OpLabel
OpBranch %12
%12 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_4;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(6, 11, /* not relevant */ 200,
100, 101, 102, 103,
/* not relevant */ 201, {}, {});
ASSERT_FALSE(transformation.IsApplicable(context.get(), fact_manager));
}
TEST(TransformationOutlineFunctionTest, DoNotOutlineIfRegionInvolvesKill) {
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%20 = OpTypeBool
%21 = OpConstantTrue %20
%3 = OpTypeFunction %2
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %6
%6 = OpLabel
OpBranch %7
%7 = OpLabel
OpSelectionMerge %10 None
OpBranchConditional %21 %8 %9
%8 = OpLabel
OpKill
%9 = OpLabel
OpBranch %10
%10 = OpLabel
OpBranch %11
%11 = OpLabel
OpBranch %12
%12 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_4;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(6, 11, /* not relevant */ 200,
100, 101, 102, 103,
/* not relevant */ 201, {}, {});
ASSERT_FALSE(transformation.IsApplicable(context.get(), fact_manager));
}
TEST(TransformationOutlineFunctionTest,
DoNotOutlineIfRegionInvolvesUnreachable) {
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%20 = OpTypeBool
%21 = OpConstantTrue %20
%3 = OpTypeFunction %2
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %6
%6 = OpLabel
OpBranch %7
%7 = OpLabel
OpSelectionMerge %10 None
OpBranchConditional %21 %8 %9
%8 = OpLabel
OpBranch %10
%9 = OpLabel
OpUnreachable
%10 = OpLabel
OpBranch %11
%11 = OpLabel
OpBranch %12
%12 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_4;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(6, 11, /* not relevant */ 200,
100, 101, 102, 103,
/* not relevant */ 201, {}, {});
ASSERT_FALSE(transformation.IsApplicable(context.get(), fact_manager));
}
TEST(TransformationOutlineFunctionTest,
DoNotOutlineIfSelectionMergeIsOutsideRegion) {
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%9 = OpTypeBool
%10 = OpConstantTrue %9
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %6
%6 = OpLabel
OpSelectionMerge %7 None
OpBranchConditional %10 %8 %7
%8 = OpLabel
OpBranch %7
%7 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_4;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(6, 8, 100, 101, 102, 103, 104,
105, {}, {});
ASSERT_FALSE(transformation.IsApplicable(context.get(), fact_manager));
}
TEST(TransformationOutlineFunctionTest, DoNotOutlineIfLoopHeadIsOutsideRegion) {
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%9 = OpTypeBool
%10 = OpConstantTrue %9
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %6
%6 = OpLabel
OpLoopMerge %8 %11 None
OpBranch %7
%7 = OpLabel
OpBranchConditional %10 %11 %8
%11 = OpLabel
OpBranch %6
%8 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_4;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(7, 8, 100, 101, 102, 103, 104,
105, {}, {});
ASSERT_FALSE(transformation.IsApplicable(context.get(), fact_manager));
}
TEST(TransformationOutlineFunctionTest,
DoNotOutlineIfLoopContinueIsOutsideRegion) {
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%9 = OpTypeBool
%10 = OpConstantTrue %9
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %6
%6 = OpLabel
OpLoopMerge %7 %8 None
OpBranch %7
%8 = OpLabel
OpBranch %6
%7 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_4;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(6, 7, 100, 101, 102, 103, 104,
105, {}, {});
ASSERT_FALSE(transformation.IsApplicable(context.get(), fact_manager));
}
TEST(TransformationOutlineFunctionTest,
DoNotOutlineWithLoopCarriedPhiDependence) {
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%9 = OpTypeBool
%10 = OpConstantTrue %9
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %6
%6 = OpLabel
%12 = OpPhi %9 %10 %5 %13 %8
OpLoopMerge %7 %8 None
OpBranch %8
%8 = OpLabel
%13 = OpCopyObject %9 %10
OpBranchConditional %10 %6 %7
%7 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_4;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(6, 7, 100, 101, 102, 103, 104,
105, {}, {});
ASSERT_FALSE(transformation.IsApplicable(context.get(), fact_manager));
}
TEST(TransformationOutlineFunctionTest,
DoNotOutlineSelectionHeaderNotInRegion) {
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%6 = OpTypeBool
%7 = OpConstantTrue %6
%4 = OpFunction %2 None %3
%5 = OpLabel
OpSelectionMerge %10 None
OpBranchConditional %7 %8 %8
%8 = OpLabel
OpBranch %9
%9 = OpLabel
OpBranch %10
%10 = OpLabel
OpBranch %11
%11 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_4;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(8, 11, 100, 101, 102, 103, 104,
105, {}, {});
ASSERT_FALSE(transformation.IsApplicable(context.get(), fact_manager));
}
TEST(TransformationOutlineFunctionTest, OutlineRegionEndingWithReturnVoid) {
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
%20 = OpTypeInt 32 0
%21 = OpConstant %20 1
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%4 = OpFunction %2 None %3
%5 = OpLabel
%22 = OpCopyObject %20 %21
OpBranch %54
%54 = OpLabel
OpBranch %57
%57 = OpLabel
%23 = OpCopyObject %20 %22
OpBranch %58
%58 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_5;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(
/*entry_block*/ 54,
/*exit_block*/ 58,
/*new_function_struct_return_type_id*/ 200,
/*new_function_type_id*/ 201,
/*new_function_id*/ 202,
/*new_function_region_entry_block*/ 203,
/*new_caller_result_id*/ 204,
/*new_callee_result_id*/ 205,
/*input_id_to_fresh_id*/ {{22, 206}},
/*output_id_to_fresh_id*/ {});
ASSERT_TRUE(transformation.IsApplicable(context.get(), fact_manager));
transformation.Apply(context.get(), &fact_manager);
ASSERT_TRUE(IsValid(env, context.get()));
std::string after_transformation = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
%20 = OpTypeInt 32 0
%21 = OpConstant %20 1
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%201 = OpTypeFunction %2 %20
%4 = OpFunction %2 None %3
%5 = OpLabel
%22 = OpCopyObject %20 %21
OpBranch %54
%54 = OpLabel
%204 = OpFunctionCall %2 %202 %22
OpReturn
OpFunctionEnd
%202 = OpFunction %2 None %201
%206 = OpFunctionParameter %20
%203 = OpLabel
OpBranch %57
%57 = OpLabel
%23 = OpCopyObject %20 %206
OpBranch %58
%58 = OpLabel
OpReturn
OpFunctionEnd
)";
ASSERT_TRUE(IsEqual(env, after_transformation, context.get()));
}
TEST(TransformationOutlineFunctionTest, OutlineRegionEndingWithReturnValue) {
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
%20 = OpTypeInt 32 0
%21 = OpConstant %20 1
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%30 = OpTypeFunction %20
%4 = OpFunction %2 None %3
%5 = OpLabel
%6 = OpFunctionCall %20 %100
OpReturn
OpFunctionEnd
%100 = OpFunction %20 None %30
%8 = OpLabel
%31 = OpCopyObject %20 %21
OpBranch %9
%9 = OpLabel
%32 = OpCopyObject %20 %31
OpBranch %10
%10 = OpLabel
OpReturnValue %32
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_5;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(
/*entry_block*/ 9,
/*exit_block*/ 10,
/*new_function_struct_return_type_id*/ 200,
/*new_function_type_id*/ 201,
/*new_function_id*/ 202,
/*new_function_region_entry_block*/ 203,
/*new_caller_result_id*/ 204,
/*new_callee_result_id*/ 205,
/*input_id_to_fresh_id*/ {{31, 206}},
/*output_id_to_fresh_id*/ {{32, 207}});
ASSERT_TRUE(transformation.IsApplicable(context.get(), fact_manager));
transformation.Apply(context.get(), &fact_manager);
ASSERT_TRUE(IsValid(env, context.get()));
std::string after_transformation = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
%20 = OpTypeInt 32 0
%21 = OpConstant %20 1
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%30 = OpTypeFunction %20
%200 = OpTypeStruct %20
%201 = OpTypeFunction %200 %20
%4 = OpFunction %2 None %3
%5 = OpLabel
%6 = OpFunctionCall %20 %100
OpReturn
OpFunctionEnd
%100 = OpFunction %20 None %30
%8 = OpLabel
%31 = OpCopyObject %20 %21
OpBranch %9
%9 = OpLabel
%204 = OpFunctionCall %200 %202 %31
%32 = OpCompositeExtract %20 %204 0
OpReturnValue %32
OpFunctionEnd
%202 = OpFunction %200 None %201
%206 = OpFunctionParameter %20
%203 = OpLabel
%207 = OpCopyObject %20 %206
OpBranch %10
%10 = OpLabel
%205 = OpCompositeConstruct %200 %207
OpReturnValue %205
OpFunctionEnd
)";
ASSERT_TRUE(IsEqual(env, after_transformation, context.get()));
}
TEST(TransformationOutlineFunctionTest,
OutlineRegionEndingWithConditionalBranch) {
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
%20 = OpTypeBool
%21 = OpConstantTrue %20
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %54
%54 = OpLabel
%6 = OpCopyObject %20 %21
OpSelectionMerge %8 None
OpBranchConditional %6 %7 %8
%7 = OpLabel
OpBranch %8
%8 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_5;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(
/*entry_block*/ 54,
/*exit_block*/ 54,
/*new_function_struct_return_type_id*/ 200,
/*new_function_type_id*/ 201,
/*new_function_id*/ 202,
/*new_function_region_entry_block*/ 203,
/*new_caller_result_id*/ 204,
/*new_callee_result_id*/ 205,
/*input_id_to_fresh_id*/ {{}},
/*output_id_to_fresh_id*/ {{6, 206}});
ASSERT_TRUE(transformation.IsApplicable(context.get(), fact_manager));
transformation.Apply(context.get(), &fact_manager);
ASSERT_TRUE(IsValid(env, context.get()));
std::string after_transformation = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
%20 = OpTypeBool
%21 = OpConstantTrue %20
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%200 = OpTypeStruct %20
%201 = OpTypeFunction %200
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %54
%54 = OpLabel
%204 = OpFunctionCall %200 %202
%6 = OpCompositeExtract %20 %204 0
OpSelectionMerge %8 None
OpBranchConditional %6 %7 %8
%7 = OpLabel
OpBranch %8
%8 = OpLabel
OpReturn
OpFunctionEnd
%202 = OpFunction %200 None %201
%203 = OpLabel
%206 = OpCopyObject %20 %21
%205 = OpCompositeConstruct %200 %206
OpReturnValue %205
OpFunctionEnd
)";
ASSERT_TRUE(IsEqual(env, after_transformation, context.get()));
}
TEST(TransformationOutlineFunctionTest,
OutlineRegionEndingWithConditionalBranch2) {
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
%20 = OpTypeBool
%21 = OpConstantTrue %20
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%4 = OpFunction %2 None %3
%5 = OpLabel
%6 = OpCopyObject %20 %21
OpBranch %54
%54 = OpLabel
OpSelectionMerge %8 None
OpBranchConditional %6 %7 %8
%7 = OpLabel
OpBranch %8
%8 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_5;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(
/*entry_block*/ 54,
/*exit_block*/ 54,
/*new_function_struct_return_type_id*/ 200,
/*new_function_type_id*/ 201,
/*new_function_id*/ 202,
/*new_function_region_entry_block*/ 203,
/*new_caller_result_id*/ 204,
/*new_callee_result_id*/ 205,
/*input_id_to_fresh_id*/ {},
/*output_id_to_fresh_id*/ {});
ASSERT_TRUE(transformation.IsApplicable(context.get(), fact_manager));
transformation.Apply(context.get(), &fact_manager);
ASSERT_TRUE(IsValid(env, context.get()));
std::string after_transformation = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
%20 = OpTypeBool
%21 = OpConstantTrue %20
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%4 = OpFunction %2 None %3
%5 = OpLabel
%6 = OpCopyObject %20 %21
OpBranch %54
%54 = OpLabel
%204 = OpFunctionCall %2 %202
OpSelectionMerge %8 None
OpBranchConditional %6 %7 %8
%7 = OpLabel
OpBranch %8
%8 = OpLabel
OpReturn
OpFunctionEnd
%202 = OpFunction %2 None %3
%203 = OpLabel
OpReturn
OpFunctionEnd
)";
ASSERT_TRUE(IsEqual(env, after_transformation, context.get()));
}
TEST(TransformationOutlineFunctionTest, DoNotOutlineRegionThatStartsWithOpPhi) {
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%6 = OpTypeBool
%7 = OpConstantTrue %6
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %21
%21 = OpLabel
%22 = OpPhi %6 %7 %5
%23 = OpCopyObject %6 %22
OpBranch %24
%24 = OpLabel
%25 = OpCopyObject %6 %23
%26 = OpCopyObject %6 %22
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_5;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(
/*entry_block*/ 21,
/*exit_block*/ 21,
/*new_function_struct_return_type_id*/ 200,
/*new_function_type_id*/ 201,
/*new_function_id*/ 202,
/*new_function_region_entry_block*/ 204,
/*new_caller_result_id*/ 205,
/*new_callee_result_id*/ 206,
/*input_id_to_fresh_id*/ {{22, 207}},
/*output_id_to_fresh_id*/ {{23, 208}});
ASSERT_FALSE(transformation.IsApplicable(context.get(), fact_manager));
}
TEST(TransformationOutlineFunctionTest,
DoNotOutlineRegionThatStartsWithLoopHeader) {
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%6 = OpTypeBool
%7 = OpConstantTrue %6
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %21
%21 = OpLabel
OpLoopMerge %22 %23 None
OpBranch %24
%24 = OpLabel
OpBranchConditional %7 %22 %23
%23 = OpLabel
OpBranch %21
%22 = OpLabel
OpBranch %25
%25 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_5;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(
/*entry_block*/ 21,
/*exit_block*/ 24,
/*new_function_struct_return_type_id*/ 200,
/*new_function_type_id*/ 201,
/*new_function_id*/ 202,
/*new_function_region_entry_block*/ 204,
/*new_caller_result_id*/ 205,
/*new_callee_result_id*/ 206,
/*input_id_to_fresh_id*/ {},
/*output_id_to_fresh_id*/ {});
ASSERT_FALSE(transformation.IsApplicable(context.get(), fact_manager));
}
TEST(TransformationOutlineFunctionTest,
DoNotOutlineRegionThatEndsWithLoopMerge) {
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%6 = OpTypeBool
%7 = OpConstantTrue %6
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %21
%21 = OpLabel
OpLoopMerge %22 %23 None
OpBranch %24
%24 = OpLabel
OpBranchConditional %7 %22 %23
%23 = OpLabel
OpBranch %21
%22 = OpLabel
OpBranch %25
%25 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_5;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(
/*entry_block*/ 5,
/*exit_block*/ 22,
/*new_function_struct_return_type_id*/ 200,
/*new_function_type_id*/ 201,
/*new_function_id*/ 202,
/*new_function_region_entry_block*/ 204,
/*new_caller_result_id*/ 205,
/*new_callee_result_id*/ 206,
/*input_id_to_fresh_id*/ {},
/*output_id_to_fresh_id*/ {});
ASSERT_FALSE(transformation.IsApplicable(context.get(), fact_manager));
}
TEST(TransformationOutlineFunctionTest, DoNotOutlineRegionThatUsesAccessChain) {
// An access chain result is a pointer, but it cannot be passed as a function
// parameter, as it is not a memory object.
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%6 = OpTypeFloat 32
%7 = OpTypeVector %6 4
%8 = OpTypePointer Function %7
%9 = OpTypePointer Function %6
%18 = OpTypeInt 32 0
%19 = OpConstant %18 0
%4 = OpFunction %2 None %3
%5 = OpLabel
%10 = OpVariable %8 Function
OpBranch %11
%11 = OpLabel
%12 = OpAccessChain %9 %10 %19
OpBranch %13
%13 = OpLabel
%14 = OpLoad %6 %12
OpBranch %15
%15 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_5;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(
/*entry_block*/ 13,
/*exit_block*/ 15,
/*new_function_struct_return_type_id*/ 200,
/*new_function_type_id*/ 201,
/*new_function_id*/ 202,
/*new_function_region_entry_block*/ 204,
/*new_caller_result_id*/ 205,
/*new_callee_result_id*/ 206,
/*input_id_to_fresh_id*/ {{12, 207}},
/*output_id_to_fresh_id*/ {});
ASSERT_FALSE(transformation.IsApplicable(context.get(), fact_manager));
}
TEST(TransformationOutlineFunctionTest, Miscellaneous1) {
// This tests outlining of some non-trivial code.
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main" %85
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
OpName %28 "buf"
OpMemberName %28 0 "u1"
OpMemberName %28 1 "u2"
OpName %30 ""
OpName %85 "color"
OpMemberDecorate %28 0 Offset 0
OpMemberDecorate %28 1 Offset 4
OpDecorate %28 Block
OpDecorate %30 DescriptorSet 0
OpDecorate %30 Binding 0
OpDecorate %85 Location 0
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%6 = OpTypeFloat 32
%7 = OpTypeVector %6 4
%10 = OpConstant %6 1
%11 = OpConstant %6 2
%12 = OpConstant %6 3
%13 = OpConstant %6 4
%14 = OpConstantComposite %7 %10 %11 %12 %13
%15 = OpTypeInt 32 1
%18 = OpConstant %15 0
%28 = OpTypeStruct %6 %6
%29 = OpTypePointer Uniform %28
%30 = OpVariable %29 Uniform
%31 = OpTypePointer Uniform %6
%35 = OpTypeBool
%39 = OpConstant %15 1
%84 = OpTypePointer Output %7
%85 = OpVariable %84 Output
%114 = OpConstant %15 8
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %22
%22 = OpLabel
%103 = OpPhi %15 %18 %5 %106 %43
%102 = OpPhi %7 %14 %5 %107 %43
%101 = OpPhi %15 %18 %5 %40 %43
%32 = OpAccessChain %31 %30 %18
%33 = OpLoad %6 %32
%34 = OpConvertFToS %15 %33
%36 = OpSLessThan %35 %101 %34
OpLoopMerge %24 %43 None
OpBranchConditional %36 %23 %24
%23 = OpLabel
%40 = OpIAdd %15 %101 %39
OpBranch %150
%150 = OpLabel
OpBranch %41
%41 = OpLabel
%107 = OpPhi %7 %102 %150 %111 %65
%106 = OpPhi %15 %103 %150 %110 %65
%104 = OpPhi %15 %40 %150 %81 %65
%47 = OpAccessChain %31 %30 %39
%48 = OpLoad %6 %47
%49 = OpConvertFToS %15 %48
%50 = OpSLessThan %35 %104 %49
OpLoopMerge %1000 %65 None
OpBranchConditional %50 %42 %1000
%42 = OpLabel
%60 = OpIAdd %15 %106 %114
%63 = OpSGreaterThan %35 %104 %60
OpBranchConditional %63 %64 %65
%64 = OpLabel
%71 = OpCompositeExtract %6 %107 0
%72 = OpFAdd %6 %71 %11
%97 = OpCompositeInsert %7 %72 %107 0
%76 = OpCompositeExtract %6 %107 3
%77 = OpConvertFToS %15 %76
%79 = OpIAdd %15 %60 %77
OpBranch %65
%65 = OpLabel
%111 = OpPhi %7 %107 %42 %97 %64
%110 = OpPhi %15 %60 %42 %79 %64
%81 = OpIAdd %15 %104 %39
OpBranch %41
%1000 = OpLabel
OpBranch %1001
%1001 = OpLabel
OpBranch %43
%43 = OpLabel
OpBranch %22
%24 = OpLabel
%87 = OpCompositeExtract %6 %102 0
%91 = OpConvertSToF %6 %103
%92 = OpCompositeConstruct %7 %87 %11 %91 %10
OpStore %85 %92
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_3;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(
/*entry_block*/ 150,
/*exit_block*/ 1001,
/*new_function_struct_return_type_id*/ 200,
/*new_function_type_id*/ 201,
/*new_function_id*/ 202,
/*new_function_region_entry_block*/ 203,
/*new_caller_result_id*/ 204,
/*new_callee_result_id*/ 205,
/*input_id_to_fresh_id*/ {{102, 300}, {103, 301}, {40, 302}},
/*output_id_to_fresh_id*/ {{106, 400}, {107, 401}});
ASSERT_TRUE(transformation.IsApplicable(context.get(), fact_manager));
transformation.Apply(context.get(), &fact_manager);
ASSERT_TRUE(IsValid(env, context.get()));
std::string after_transformation = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main" %85
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
OpName %4 "main"
OpName %28 "buf"
OpMemberName %28 0 "u1"
OpMemberName %28 1 "u2"
OpName %30 ""
OpName %85 "color"
OpMemberDecorate %28 0 Offset 0
OpMemberDecorate %28 1 Offset 4
OpDecorate %28 Block
OpDecorate %30 DescriptorSet 0
OpDecorate %30 Binding 0
OpDecorate %85 Location 0
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%6 = OpTypeFloat 32
%7 = OpTypeVector %6 4
%10 = OpConstant %6 1
%11 = OpConstant %6 2
%12 = OpConstant %6 3
%13 = OpConstant %6 4
%14 = OpConstantComposite %7 %10 %11 %12 %13
%15 = OpTypeInt 32 1
%18 = OpConstant %15 0
%28 = OpTypeStruct %6 %6
%29 = OpTypePointer Uniform %28
%30 = OpVariable %29 Uniform
%31 = OpTypePointer Uniform %6
%35 = OpTypeBool
%39 = OpConstant %15 1
%84 = OpTypePointer Output %7
%85 = OpVariable %84 Output
%114 = OpConstant %15 8
%200 = OpTypeStruct %7 %15
%201 = OpTypeFunction %200 %15 %7 %15
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %22
%22 = OpLabel
%103 = OpPhi %15 %18 %5 %106 %43
%102 = OpPhi %7 %14 %5 %107 %43
%101 = OpPhi %15 %18 %5 %40 %43
%32 = OpAccessChain %31 %30 %18
%33 = OpLoad %6 %32
%34 = OpConvertFToS %15 %33
%36 = OpSLessThan %35 %101 %34
OpLoopMerge %24 %43 None
OpBranchConditional %36 %23 %24
%23 = OpLabel
%40 = OpIAdd %15 %101 %39
OpBranch %150
%150 = OpLabel
%204 = OpFunctionCall %200 %202 %103 %102 %40
%107 = OpCompositeExtract %7 %204 0
%106 = OpCompositeExtract %15 %204 1
OpBranch %43
%43 = OpLabel
OpBranch %22
%24 = OpLabel
%87 = OpCompositeExtract %6 %102 0
%91 = OpConvertSToF %6 %103
%92 = OpCompositeConstruct %7 %87 %11 %91 %10
OpStore %85 %92
OpReturn
OpFunctionEnd
%202 = OpFunction %200 None %201
%301 = OpFunctionParameter %15
%300 = OpFunctionParameter %7
%302 = OpFunctionParameter %15
%203 = OpLabel
OpBranch %41
%41 = OpLabel
%401 = OpPhi %7 %300 %203 %111 %65
%400 = OpPhi %15 %301 %203 %110 %65
%104 = OpPhi %15 %302 %203 %81 %65
%47 = OpAccessChain %31 %30 %39
%48 = OpLoad %6 %47
%49 = OpConvertFToS %15 %48
%50 = OpSLessThan %35 %104 %49
OpLoopMerge %1000 %65 None
OpBranchConditional %50 %42 %1000
%42 = OpLabel
%60 = OpIAdd %15 %400 %114
%63 = OpSGreaterThan %35 %104 %60
OpBranchConditional %63 %64 %65
%64 = OpLabel
%71 = OpCompositeExtract %6 %401 0
%72 = OpFAdd %6 %71 %11
%97 = OpCompositeInsert %7 %72 %401 0
%76 = OpCompositeExtract %6 %401 3
%77 = OpConvertFToS %15 %76
%79 = OpIAdd %15 %60 %77
OpBranch %65
%65 = OpLabel
%111 = OpPhi %7 %401 %42 %97 %64
%110 = OpPhi %15 %60 %42 %79 %64
%81 = OpIAdd %15 %104 %39
OpBranch %41
%1000 = OpLabel
OpBranch %1001
%1001 = OpLabel
%205 = OpCompositeConstruct %200 %401 %400
OpReturnValue %205
OpFunctionEnd
)";
ASSERT_TRUE(IsEqual(env, after_transformation, context.get()));
}
TEST(TransformationOutlineFunctionTest, Miscellaneous2) {
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
OpSource ESSL 310
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%21 = OpTypeBool
%167 = OpConstantTrue %21
%168 = OpConstantFalse %21
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %34
%34 = OpLabel
OpLoopMerge %36 %37 None
OpBranchConditional %168 %37 %38
%38 = OpLabel
OpBranchConditional %168 %37 %36
%37 = OpLabel
OpBranch %34
%36 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_5;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(
/*entry_block*/ 38,
/*exit_block*/ 36,
/*new_function_struct_return_type_id*/ 200,
/*new_function_type_id*/ 201,
/*new_function_id*/ 202,
/*new_function_region_entry_block*/ 203,
/*new_caller_result_id*/ 204,
/*new_callee_result_id*/ 205,
/*input_id_to_fresh_id*/ {},
/*output_id_to_fresh_id*/ {});
ASSERT_FALSE(transformation.IsApplicable(context.get(), fact_manager));
}
TEST(TransformationOutlineFunctionTest, Miscellaneous3) {
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %6 "main"
OpExecutionMode %6 OriginUpperLeft
OpSource ESSL 310
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%21 = OpTypeBool
%167 = OpConstantTrue %21
%6 = OpFunction %2 None %3
%7 = OpLabel
OpBranch %80
%80 = OpLabel
OpBranch %14
%14 = OpLabel
OpLoopMerge %16 %17 None
OpBranch %18
%18 = OpLabel
OpBranchConditional %167 %15 %16
%15 = OpLabel
OpBranch %17
%16 = OpLabel
OpBranch %81
%81 = OpLabel
OpReturn
%17 = OpLabel
OpBranch %14
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_5;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
ASSERT_TRUE(IsValid(env, context.get()));
FactManager fact_manager;
TransformationOutlineFunction transformation(
/*entry_block*/ 80,
/*exit_block*/ 81,
/*new_function_struct_return_type_id*/ 300,
/*new_function_type_id*/ 301,
/*new_function_id*/ 302,
/*new_function_region_entry_block*/ 304,
/*new_caller_result_id*/ 305,
/*new_callee_result_id*/ 306,
/*input_id_to_fresh_id*/ {},
/*output_id_to_fresh_id*/ {});
ASSERT_TRUE(transformation.IsApplicable(context.get(), fact_manager));
transformation.Apply(context.get(), &fact_manager);
ASSERT_TRUE(IsValid(env, context.get()));
std::string after_transformation = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %6 "main"
OpExecutionMode %6 OriginUpperLeft
OpSource ESSL 310
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%21 = OpTypeBool
%167 = OpConstantTrue %21
%6 = OpFunction %2 None %3
%7 = OpLabel
OpBranch %80
%80 = OpLabel
%305 = OpFunctionCall %2 %302
OpReturn
OpFunctionEnd
%302 = OpFunction %2 None %3
%304 = OpLabel
OpBranch %14
%14 = OpLabel
OpLoopMerge %16 %17 None
OpBranch %18
%18 = OpLabel
OpBranchConditional %167 %15 %16
%15 = OpLabel
OpBranch %17
%16 = OpLabel
OpBranch %81
%81 = OpLabel
OpReturn
%17 = OpLabel
OpBranch %14
OpFunctionEnd
)";
ASSERT_TRUE(IsEqual(env, after_transformation, context.get()));
}
} // namespace
} // namespace fuzz
} // namespace spvtools