SPIRV-Tools/test/fuzz/transformation_inline_function_test.cpp
2020-10-19 09:41:01 +01:00

1117 lines
39 KiB
C++

// Copyright (c) 2020 André Perez Maselco
//
// 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_inline_function.h"
#include "gtest/gtest.h"
#include "source/fuzz/counter_overflow_id_source.h"
#include "source/fuzz/fuzzer_util.h"
#include "source/fuzz/instruction_descriptor.h"
#include "test/fuzz/fuzz_test_util.h"
namespace spvtools {
namespace fuzz {
namespace {
TEST(TransformationInlineFunctionTest, IsApplicable) {
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %52 "main"
OpExecutionMode %52 OriginUpperLeft
OpName %56 "function_with_void_return"
; Types
%2 = OpTypeBool
%3 = OpTypeFloat 32
%4 = OpTypeVector %3 4
%5 = OpTypePointer Function %4
%6 = OpTypeVoid
%7 = OpTypeFunction %6
%8 = OpTypeFunction %3 %5 %5
; Constant scalars
%9 = OpConstant %3 1
%10 = OpConstant %3 2
%11 = OpConstant %3 3
%12 = OpConstant %3 4
%13 = OpConstant %3 5
%14 = OpConstant %3 6
%15 = OpConstant %3 7
%16 = OpConstant %3 8
%17 = OpConstantTrue %2
; Constant vectors
%18 = OpConstantComposite %4 %9 %10 %11 %12
%19 = OpConstantComposite %4 %13 %14 %15 %16
; function with void return
%20 = OpFunction %6 None %7
%21 = OpLabel
OpReturn
OpFunctionEnd
; function with early return
%22 = OpFunction %6 None %7
%23 = OpLabel
OpSelectionMerge %26 None
OpBranchConditional %17 %24 %25
%24 = OpLabel
OpReturn
%25 = OpLabel
OpBranch %26
%26 = OpLabel
OpReturn
OpFunctionEnd
; function containing an OpKill instruction
%27 = OpFunction %6 None %7
%28 = OpLabel
OpKill
OpFunctionEnd
; function containing an OpUnreachable instruction
%29 = OpFunction %6 None %7
%30 = OpLabel
OpUnreachable
OpFunctionEnd
; dot product function
%31 = OpFunction %3 None %8
%32 = OpFunctionParameter %5
%33 = OpFunctionParameter %5
%34 = OpLabel
%35 = OpLoad %4 %32
%36 = OpLoad %4 %33
%37 = OpCompositeExtract %3 %35 0
%38 = OpCompositeExtract %3 %36 0
%39 = OpFMul %3 %37 %38
%40 = OpCompositeExtract %3 %35 1
%41 = OpCompositeExtract %3 %36 1
%42 = OpFMul %3 %40 %41
%43 = OpCompositeExtract %3 %35 2
%44 = OpCompositeExtract %3 %36 2
%45 = OpFMul %3 %43 %44
%46 = OpCompositeExtract %3 %35 3
%47 = OpCompositeExtract %3 %36 3
%48 = OpFMul %3 %46 %47
%49 = OpFAdd %3 %39 %42
%50 = OpFAdd %3 %45 %49
%51 = OpFAdd %3 %48 %50
OpReturnValue %51
OpFunctionEnd
; main function
%52 = OpFunction %6 None %7
%53 = OpLabel
%54 = OpVariable %5 Function
%55 = OpVariable %5 Function
%56 = OpFunctionCall %6 %20 ; function with void return
OpBranch %57
%57 = OpLabel
%59 = OpFunctionCall %6 %22 ; function with early return
OpBranch %60
%60 = OpLabel
%61 = OpFunctionCall %6 %27 ; function containing OpKill
OpBranch %62
%62 = OpLabel
%63 = OpFunctionCall %6 %29 ; function containing OpUnreachable
OpBranch %64
%64 = OpLabel
OpStore %54 %18
OpStore %55 %19
%65 = OpFunctionCall %3 %31 %54 %55 ; dot product function
OpBranch %66
%66 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_5;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
spvtools::ValidatorOptions validator_options;
ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options,
kConsoleMessageConsumer));
TransformationContext transformation_context(
MakeUnique<FactManager>(context.get()), validator_options);
// Tests undefined OpFunctionCall instruction.
auto transformation = TransformationInlineFunction(67, {});
ASSERT_FALSE(
transformation.IsApplicable(context.get(), transformation_context));
// Tests false OpFunctionCall instruction.
transformation = TransformationInlineFunction(42, {});
ASSERT_FALSE(
transformation.IsApplicable(context.get(), transformation_context));
// Tests use of called function with void return.
transformation = TransformationInlineFunction(56, {});
ASSERT_FALSE(
transformation.IsApplicable(context.get(), transformation_context));
// Tests called function having an early return.
transformation =
TransformationInlineFunction(59, {{24, 67}, {25, 68}, {26, 69}});
ASSERT_FALSE(
transformation.IsApplicable(context.get(), transformation_context));
// Tests called function containing an OpKill instruction.
transformation = TransformationInlineFunction(61, {});
ASSERT_FALSE(
transformation.IsApplicable(context.get(), transformation_context));
// Tests called function containing an OpUnreachable instruction.
transformation = TransformationInlineFunction(63, {});
ASSERT_FALSE(
transformation.IsApplicable(context.get(), transformation_context));
// Tests applicable transformation.
transformation = TransformationInlineFunction(65, {{35, 67},
{36, 68},
{37, 69},
{38, 70},
{39, 71},
{40, 72},
{41, 73},
{42, 74},
{43, 75},
{44, 76},
{45, 77},
{46, 78},
{47, 79},
{48, 80},
{49, 81},
{50, 82},
{51, 83}});
ASSERT_TRUE(
transformation.IsApplicable(context.get(), transformation_context));
}
TEST(TransformationInlineFunctionTest, Apply) {
std::string reference_shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Vertex %39 "main"
; Types
%2 = OpTypeFloat 32
%3 = OpTypeVector %2 4
%4 = OpTypePointer Function %3
%5 = OpTypeVoid
%6 = OpTypeFunction %5
%7 = OpTypeFunction %2 %4 %4
; Constant scalars
%8 = OpConstant %2 1
%9 = OpConstant %2 2
%10 = OpConstant %2 3
%11 = OpConstant %2 4
%12 = OpConstant %2 5
%13 = OpConstant %2 6
%14 = OpConstant %2 7
%15 = OpConstant %2 8
; Constant vectors
%16 = OpConstantComposite %3 %8 %9 %10 %11
%17 = OpConstantComposite %3 %12 %13 %14 %15
; dot product function
%18 = OpFunction %2 None %7
%19 = OpFunctionParameter %4
%20 = OpFunctionParameter %4
%21 = OpLabel
%22 = OpLoad %3 %19
%23 = OpLoad %3 %20
%24 = OpCompositeExtract %2 %22 0
%25 = OpCompositeExtract %2 %23 0
%26 = OpFMul %2 %24 %25
%27 = OpCompositeExtract %2 %22 1
%28 = OpCompositeExtract %2 %23 1
%29 = OpFMul %2 %27 %28
%30 = OpCompositeExtract %2 %22 2
%31 = OpCompositeExtract %2 %23 2
%32 = OpFMul %2 %30 %31
%33 = OpCompositeExtract %2 %22 3
%34 = OpCompositeExtract %2 %23 3
%35 = OpFMul %2 %33 %34
%36 = OpFAdd %2 %26 %29
%37 = OpFAdd %2 %32 %36
%38 = OpFAdd %2 %35 %37
OpReturnValue %38
OpFunctionEnd
; main function
%39 = OpFunction %5 None %6
%40 = OpLabel
%41 = OpVariable %4 Function
%42 = OpVariable %4 Function
OpStore %41 %16
OpStore %42 %17
%43 = OpFunctionCall %2 %18 %41 %42 ; dot product function call
OpBranch %44
%44 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_5;
const auto consumer = nullptr;
const auto context =
BuildModule(env, consumer, reference_shader, kFuzzAssembleOption);
spvtools::ValidatorOptions validator_options;
ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options,
kConsoleMessageConsumer));
TransformationContext transformation_context(
MakeUnique<FactManager>(context.get()), validator_options);
auto transformation = TransformationInlineFunction(43, {{22, 45},
{23, 46},
{24, 47},
{25, 48},
{26, 49},
{27, 50},
{28, 51},
{29, 52},
{30, 53},
{31, 54},
{32, 55},
{33, 56},
{34, 57},
{35, 58},
{36, 59},
{37, 60},
{38, 61}});
ApplyAndCheckFreshIds(transformation, context.get(), &transformation_context);
std::string variant_shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Vertex %39 "main"
; Types
%2 = OpTypeFloat 32
%3 = OpTypeVector %2 4
%4 = OpTypePointer Function %3
%5 = OpTypeVoid
%6 = OpTypeFunction %5
%7 = OpTypeFunction %2 %4 %4
; Constant scalars
%8 = OpConstant %2 1
%9 = OpConstant %2 2
%10 = OpConstant %2 3
%11 = OpConstant %2 4
%12 = OpConstant %2 5
%13 = OpConstant %2 6
%14 = OpConstant %2 7
%15 = OpConstant %2 8
; Constant vectors
%16 = OpConstantComposite %3 %8 %9 %10 %11
%17 = OpConstantComposite %3 %12 %13 %14 %15
; dot product function
%18 = OpFunction %2 None %7
%19 = OpFunctionParameter %4
%20 = OpFunctionParameter %4
%21 = OpLabel
%22 = OpLoad %3 %19
%23 = OpLoad %3 %20
%24 = OpCompositeExtract %2 %22 0
%25 = OpCompositeExtract %2 %23 0
%26 = OpFMul %2 %24 %25
%27 = OpCompositeExtract %2 %22 1
%28 = OpCompositeExtract %2 %23 1
%29 = OpFMul %2 %27 %28
%30 = OpCompositeExtract %2 %22 2
%31 = OpCompositeExtract %2 %23 2
%32 = OpFMul %2 %30 %31
%33 = OpCompositeExtract %2 %22 3
%34 = OpCompositeExtract %2 %23 3
%35 = OpFMul %2 %33 %34
%36 = OpFAdd %2 %26 %29
%37 = OpFAdd %2 %32 %36
%38 = OpFAdd %2 %35 %37
OpReturnValue %38
OpFunctionEnd
; main function
%39 = OpFunction %5 None %6
%40 = OpLabel
%41 = OpVariable %4 Function
%42 = OpVariable %4 Function
OpStore %41 %16
OpStore %42 %17
%45 = OpLoad %3 %41
%46 = OpLoad %3 %42
%47 = OpCompositeExtract %2 %45 0
%48 = OpCompositeExtract %2 %46 0
%49 = OpFMul %2 %47 %48
%50 = OpCompositeExtract %2 %45 1
%51 = OpCompositeExtract %2 %46 1
%52 = OpFMul %2 %50 %51
%53 = OpCompositeExtract %2 %45 2
%54 = OpCompositeExtract %2 %46 2
%55 = OpFMul %2 %53 %54
%56 = OpCompositeExtract %2 %45 3
%57 = OpCompositeExtract %2 %46 3
%58 = OpFMul %2 %56 %57
%59 = OpFAdd %2 %49 %52
%60 = OpFAdd %2 %55 %59
%61 = OpFAdd %2 %58 %60
%43 = OpCopyObject %2 %61
OpBranch %44
%44 = OpLabel
OpReturn
OpFunctionEnd
)";
ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options,
kConsoleMessageConsumer));
ASSERT_TRUE(IsEqual(env, variant_shader, context.get()));
}
TEST(TransformationInlineFunctionTest, ApplyToMultipleFunctions) {
std::string reference_shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Vertex %15 "main"
; Types
%2 = OpTypeInt 32 1
%3 = OpTypeBool
%4 = OpTypePointer Private %2
%5 = OpTypePointer Function %2
%6 = OpTypeVoid
%7 = OpTypeFunction %6
%8 = OpTypeFunction %2 %5
%9 = OpTypeFunction %2 %2
; Constants
%10 = OpConstant %2 0
%11 = OpConstant %2 1
%12 = OpConstant %2 2
%13 = OpConstant %2 3
; Global variable
%14 = OpVariable %4 Private
; main function
%15 = OpFunction %6 None %7
%16 = OpLabel
%17 = OpVariable %5 Function
%18 = OpVariable %5 Function
%19 = OpVariable %5 Function
OpStore %17 %13
%20 = OpLoad %2 %17
OpStore %18 %20
%21 = OpFunctionCall %2 %36 %18
OpBranch %22
%22 = OpLabel
%23 = OpFunctionCall %2 %36 %18
OpStore %17 %21
%24 = OpLoad %2 %17
%25 = OpFunctionCall %2 %54 %24
OpBranch %26
%26 = OpLabel
%27 = OpFunctionCall %2 %54 %24
%28 = OpLoad %2 %17
%29 = OpIAdd %2 %28 %25
OpStore %17 %29
%30 = OpFunctionCall %6 %67
OpBranch %31
%31 = OpLabel
%32 = OpFunctionCall %6 %67
%33 = OpLoad %2 %14
%34 = OpLoad %2 %17
%35 = OpIAdd %2 %34 %33
OpStore %17 %35
OpReturn
OpFunctionEnd
; Function %36
%36 = OpFunction %2 None %8
%37 = OpFunctionParameter %5
%38 = OpLabel
%39 = OpVariable %5 Function
%40 = OpVariable %5 Function
OpStore %39 %10
OpBranch %41
%41 = OpLabel
OpLoopMerge %52 %49 None
OpBranch %42
%42 = OpLabel
%43 = OpLoad %2 %39
%44 = OpLoad %2 %37
%45 = OpSLessThan %3 %43 %44
OpBranchConditional %45 %46 %52
%46 = OpLabel
%47 = OpLoad %2 %40
%48 = OpIAdd %2 %47 %11
OpStore %40 %48
OpBranch %49
%49 = OpLabel
%50 = OpLoad %2 %39
%51 = OpIAdd %2 %50 %12
OpStore %39 %51
OpBranch %41
%52 = OpLabel
%53 = OpLoad %2 %40
OpReturnValue %53
OpFunctionEnd
; Function %54
%54 = OpFunction %2 None %9
%55 = OpFunctionParameter %2
%56 = OpLabel
%57 = OpVariable %5 Function
OpStore %57 %10
%58 = OpSGreaterThan %3 %55 %10
OpSelectionMerge %62 None
OpBranchConditional %58 %64 %59
%59 = OpLabel
%60 = OpLoad %2 %57
%61 = OpISub %2 %60 %12
OpStore %57 %61
OpBranch %62
%62 = OpLabel
%63 = OpLoad %2 %57
OpReturnValue %63
%64 = OpLabel
%65 = OpLoad %2 %57
%66 = OpIAdd %2 %65 %11
OpStore %57 %66
OpBranch %62
OpFunctionEnd
; Function %67
%67 = OpFunction %6 None %7
%68 = OpLabel
OpStore %14 %12
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_3;
const auto consumer = nullptr;
const auto context =
BuildModule(env, consumer, reference_shader, kFuzzAssembleOption);
spvtools::ValidatorOptions validator_options;
ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options,
kConsoleMessageConsumer));
TransformationContext transformation_context(
MakeUnique<FactManager>(context.get()), validator_options);
auto transformation = TransformationInlineFunction(30, {});
ASSERT_TRUE(
transformation.IsApplicable(context.get(), transformation_context));
ApplyAndCheckFreshIds(transformation, context.get(), &transformation_context);
// Tests a parameter included in the id map.
transformation = TransformationInlineFunction(25, {{55, 69},
{56, 70},
{57, 71},
{58, 72},
{59, 73},
{60, 74},
{61, 75},
{62, 76},
{63, 77},
{64, 78},
{65, 79},
{66, 80}});
ASSERT_FALSE(
transformation.IsApplicable(context.get(), transformation_context));
#ifndef NDEBUG
// Tests the id of the returned value not included in the id map.
transformation = TransformationInlineFunction(25, {{56, 69},
{57, 70},
{58, 71},
{59, 72},
{60, 73},
{61, 74},
{62, 75},
{64, 76},
{65, 77},
{66, 78}});
ASSERT_DEATH(
transformation.IsApplicable(context.get(), transformation_context),
"Bad attempt to query whether overflow ids are available.");
#endif
transformation = TransformationInlineFunction(25, {{57, 69},
{58, 70},
{59, 71},
{60, 72},
{61, 73},
{62, 74},
{63, 75},
{64, 76},
{65, 77},
{66, 78}});
ASSERT_TRUE(
transformation.IsApplicable(context.get(), transformation_context));
ApplyAndCheckFreshIds(transformation, context.get(), &transformation_context);
transformation = TransformationInlineFunction(21, {{39, 79},
{40, 80},
{41, 81},
{42, 82},
{43, 83},
{44, 84},
{45, 85},
{46, 86},
{47, 87},
{48, 88},
{49, 89},
{50, 90},
{51, 91},
{52, 92},
{53, 93}});
ASSERT_TRUE(
transformation.IsApplicable(context.get(), transformation_context));
ApplyAndCheckFreshIds(transformation, context.get(), &transformation_context);
std::string variant_shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Vertex %15 "main"
; Types
%2 = OpTypeInt 32 1
%3 = OpTypeBool
%4 = OpTypePointer Private %2
%5 = OpTypePointer Function %2
%6 = OpTypeVoid
%7 = OpTypeFunction %6
%8 = OpTypeFunction %2 %5
%9 = OpTypeFunction %2 %2
; Constants
%10 = OpConstant %2 0
%11 = OpConstant %2 1
%12 = OpConstant %2 2
%13 = OpConstant %2 3
; Global variable
%14 = OpVariable %4 Private
; main function
%15 = OpFunction %6 None %7
%16 = OpLabel
%80 = OpVariable %5 Function
%79 = OpVariable %5 Function
%69 = OpVariable %5 Function
%17 = OpVariable %5 Function
%18 = OpVariable %5 Function
%19 = OpVariable %5 Function
OpStore %17 %13
%20 = OpLoad %2 %17
OpStore %18 %20
OpStore %79 %10
OpBranch %81
%81 = OpLabel
OpLoopMerge %92 %89 None
OpBranch %82
%82 = OpLabel
%83 = OpLoad %2 %79
%84 = OpLoad %2 %18
%85 = OpSLessThan %3 %83 %84
OpBranchConditional %85 %86 %92
%86 = OpLabel
%87 = OpLoad %2 %80
%88 = OpIAdd %2 %87 %11
OpStore %80 %88
OpBranch %89
%89 = OpLabel
%90 = OpLoad %2 %79
%91 = OpIAdd %2 %90 %12
OpStore %79 %91
OpBranch %81
%92 = OpLabel
%93 = OpLoad %2 %80
%21 = OpCopyObject %2 %93
OpBranch %22
%22 = OpLabel
%23 = OpFunctionCall %2 %36 %18
OpStore %17 %21
%24 = OpLoad %2 %17
OpStore %69 %10
%70 = OpSGreaterThan %3 %24 %10
OpSelectionMerge %74 None
OpBranchConditional %70 %76 %71
%71 = OpLabel
%72 = OpLoad %2 %69
%73 = OpISub %2 %72 %12
OpStore %69 %73
OpBranch %74
%74 = OpLabel
%75 = OpLoad %2 %69
%25 = OpCopyObject %2 %75
OpBranch %26
%76 = OpLabel
%77 = OpLoad %2 %69
%78 = OpIAdd %2 %77 %11
OpStore %69 %78
OpBranch %74
%26 = OpLabel
%27 = OpFunctionCall %2 %54 %24
%28 = OpLoad %2 %17
%29 = OpIAdd %2 %28 %25
OpStore %17 %29
OpStore %14 %12
OpBranch %31
%31 = OpLabel
%32 = OpFunctionCall %6 %67
%33 = OpLoad %2 %14
%34 = OpLoad %2 %17
%35 = OpIAdd %2 %34 %33
OpStore %17 %35
OpReturn
OpFunctionEnd
; Function %36
%36 = OpFunction %2 None %8
%37 = OpFunctionParameter %5
%38 = OpLabel
%39 = OpVariable %5 Function
%40 = OpVariable %5 Function
OpStore %39 %10
OpBranch %41
%41 = OpLabel
OpLoopMerge %52 %49 None
OpBranch %42
%42 = OpLabel
%43 = OpLoad %2 %39
%44 = OpLoad %2 %37
%45 = OpSLessThan %3 %43 %44
OpBranchConditional %45 %46 %52
%46 = OpLabel
%47 = OpLoad %2 %40
%48 = OpIAdd %2 %47 %11
OpStore %40 %48
OpBranch %49
%49 = OpLabel
%50 = OpLoad %2 %39
%51 = OpIAdd %2 %50 %12
OpStore %39 %51
OpBranch %41
%52 = OpLabel
%53 = OpLoad %2 %40
OpReturnValue %53
OpFunctionEnd
; Function %54
%54 = OpFunction %2 None %9
%55 = OpFunctionParameter %2
%56 = OpLabel
%57 = OpVariable %5 Function
OpStore %57 %10
%58 = OpSGreaterThan %3 %55 %10
OpSelectionMerge %62 None
OpBranchConditional %58 %64 %59
%59 = OpLabel
%60 = OpLoad %2 %57
%61 = OpISub %2 %60 %12
OpStore %57 %61
OpBranch %62
%62 = OpLabel
%63 = OpLoad %2 %57
OpReturnValue %63
%64 = OpLabel
%65 = OpLoad %2 %57
%66 = OpIAdd %2 %65 %11
OpStore %57 %66
OpBranch %62
OpFunctionEnd
; Function %67
%67 = OpFunction %6 None %7
%68 = OpLabel
OpStore %14 %12
OpReturn
OpFunctionEnd
)";
ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options,
kConsoleMessageConsumer));
ASSERT_TRUE(IsEqual(env, variant_shader, context.get()));
}
TEST(TransformationInlineFunctionTest, HandlesOpPhisInTheSecondBlock) {
std::string shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%10 = OpTypeInt 32 0
%11 = OpUndef %10
%4 = OpFunction %2 None %3
%5 = OpLabel
%6 = OpFunctionCall %2 %7
OpBranch %14
%14 = OpLabel
OpReturn
OpFunctionEnd
%7 = OpFunction %2 None %3
%8 = OpLabel
OpBranch %13
%13 = OpLabel
%12 = OpPhi %10 %11 %8
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_3;
const auto consumer = nullptr;
const auto context = BuildModule(env, consumer, shader, kFuzzAssembleOption);
spvtools::ValidatorOptions validator_options;
ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options,
kConsoleMessageConsumer));
TransformationContext transformation_context(
MakeUnique<FactManager>(context.get()), validator_options);
TransformationInlineFunction transformation(6,
{{{8, 20}, {13, 21}, {12, 22}}});
ASSERT_TRUE(
transformation.IsApplicable(context.get(), transformation_context));
ApplyAndCheckFreshIds(transformation, context.get(), &transformation_context);
ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options,
kConsoleMessageConsumer));
std::string after_transformation = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Fragment %4 "main"
OpExecutionMode %4 OriginUpperLeft
%2 = OpTypeVoid
%3 = OpTypeFunction %2
%10 = OpTypeInt 32 0
%11 = OpUndef %10
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %21
%21 = OpLabel
%22 = OpPhi %10 %11 %5
OpBranch %14
%14 = OpLabel
OpReturn
OpFunctionEnd
%7 = OpFunction %2 None %3
%8 = OpLabel
OpBranch %13
%13 = OpLabel
%12 = OpPhi %10 %11 %8
OpReturn
OpFunctionEnd
)";
ASSERT_TRUE(IsEqual(env, after_transformation, context.get()));
}
TEST(TransformationInlineFunctionTest, OverflowIds) {
std::string reference_shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Vertex %39 "main"
; Types
%2 = OpTypeFloat 32
%3 = OpTypeVector %2 4
%4 = OpTypePointer Function %3
%5 = OpTypeVoid
%6 = OpTypeFunction %5
%7 = OpTypeFunction %2 %4 %4
; Constant scalars
%8 = OpConstant %2 1
%9 = OpConstant %2 2
%10 = OpConstant %2 3
%11 = OpConstant %2 4
%12 = OpConstant %2 5
%13 = OpConstant %2 6
%14 = OpConstant %2 7
%15 = OpConstant %2 8
; Constant vectors
%16 = OpConstantComposite %3 %8 %9 %10 %11
%17 = OpConstantComposite %3 %12 %13 %14 %15
; dot product function
%18 = OpFunction %2 None %7
%19 = OpFunctionParameter %4
%20 = OpFunctionParameter %4
%21 = OpLabel
%22 = OpLoad %3 %19
%23 = OpLoad %3 %20
%24 = OpCompositeExtract %2 %22 0
%25 = OpCompositeExtract %2 %23 0
%26 = OpFMul %2 %24 %25
%27 = OpCompositeExtract %2 %22 1
%28 = OpCompositeExtract %2 %23 1
%29 = OpFMul %2 %27 %28
OpBranch %100
%100 = OpLabel
%30 = OpCompositeExtract %2 %22 2
%31 = OpCompositeExtract %2 %23 2
%32 = OpFMul %2 %30 %31
%33 = OpCompositeExtract %2 %22 3
%34 = OpCompositeExtract %2 %23 3
%35 = OpFMul %2 %33 %34
%36 = OpFAdd %2 %26 %29
%37 = OpFAdd %2 %32 %36
%38 = OpFAdd %2 %35 %37
OpReturnValue %38
OpFunctionEnd
; main function
%39 = OpFunction %5 None %6
%40 = OpLabel
%41 = OpVariable %4 Function
%42 = OpVariable %4 Function
OpStore %41 %16
OpStore %42 %17
%43 = OpFunctionCall %2 %18 %41 %42 ; dot product function call
OpBranch %44
%44 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_5;
const auto consumer = nullptr;
const auto context =
BuildModule(env, consumer, reference_shader, kFuzzAssembleOption);
spvtools::ValidatorOptions validator_options;
ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options,
kConsoleMessageConsumer));
auto overflow_ids_unique_ptr = MakeUnique<CounterOverflowIdSource>(1000);
auto overflow_ids_ptr = overflow_ids_unique_ptr.get();
TransformationContext transformation_context(
MakeUnique<FactManager>(context.get()), validator_options,
std::move(overflow_ids_unique_ptr));
auto transformation = TransformationInlineFunction(43, {{22, 45},
{23, 46},
{24, 47},
{25, 48},
{26, 49},
{27, 50},
{28, 51},
{29, 52}});
// The following ids are left un-mapped; overflow ids will be required for
// them: 30, 31, 32, 33, 34, 35, 36, 37, 38, 100
ASSERT_TRUE(
transformation.IsApplicable(context.get(), transformation_context));
ApplyAndCheckFreshIds(transformation, context.get(), &transformation_context,
overflow_ids_ptr->GetIssuedOverflowIds());
std::string variant_shader = R"(
OpCapability Shader
%1 = OpExtInstImport "GLSL.std.450"
OpMemoryModel Logical GLSL450
OpEntryPoint Vertex %39 "main"
; Types
%2 = OpTypeFloat 32
%3 = OpTypeVector %2 4
%4 = OpTypePointer Function %3
%5 = OpTypeVoid
%6 = OpTypeFunction %5
%7 = OpTypeFunction %2 %4 %4
; Constant scalars
%8 = OpConstant %2 1
%9 = OpConstant %2 2
%10 = OpConstant %2 3
%11 = OpConstant %2 4
%12 = OpConstant %2 5
%13 = OpConstant %2 6
%14 = OpConstant %2 7
%15 = OpConstant %2 8
; Constant vectors
%16 = OpConstantComposite %3 %8 %9 %10 %11
%17 = OpConstantComposite %3 %12 %13 %14 %15
; dot product function
%18 = OpFunction %2 None %7
%19 = OpFunctionParameter %4
%20 = OpFunctionParameter %4
%21 = OpLabel
%22 = OpLoad %3 %19
%23 = OpLoad %3 %20
%24 = OpCompositeExtract %2 %22 0
%25 = OpCompositeExtract %2 %23 0
%26 = OpFMul %2 %24 %25
%27 = OpCompositeExtract %2 %22 1
%28 = OpCompositeExtract %2 %23 1
%29 = OpFMul %2 %27 %28
OpBranch %100
%100 = OpLabel
%30 = OpCompositeExtract %2 %22 2
%31 = OpCompositeExtract %2 %23 2
%32 = OpFMul %2 %30 %31
%33 = OpCompositeExtract %2 %22 3
%34 = OpCompositeExtract %2 %23 3
%35 = OpFMul %2 %33 %34
%36 = OpFAdd %2 %26 %29
%37 = OpFAdd %2 %32 %36
%38 = OpFAdd %2 %35 %37
OpReturnValue %38
OpFunctionEnd
; main function
%39 = OpFunction %5 None %6
%40 = OpLabel
%41 = OpVariable %4 Function
%42 = OpVariable %4 Function
OpStore %41 %16
OpStore %42 %17
%45 = OpLoad %3 %41
%46 = OpLoad %3 %42
%47 = OpCompositeExtract %2 %45 0
%48 = OpCompositeExtract %2 %46 0
%49 = OpFMul %2 %47 %48
%50 = OpCompositeExtract %2 %45 1
%51 = OpCompositeExtract %2 %46 1
%52 = OpFMul %2 %50 %51
OpBranch %1000
%1000 = OpLabel
%1001 = OpCompositeExtract %2 %45 2
%1002 = OpCompositeExtract %2 %46 2
%1003 = OpFMul %2 %1001 %1002
%1004 = OpCompositeExtract %2 %45 3
%1005 = OpCompositeExtract %2 %46 3
%1006 = OpFMul %2 %1004 %1005
%1007 = OpFAdd %2 %49 %52
%1008 = OpFAdd %2 %1003 %1007
%1009 = OpFAdd %2 %1006 %1008
%43 = OpCopyObject %2 %1009
OpBranch %44
%44 = OpLabel
OpReturn
OpFunctionEnd
)";
ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options,
kConsoleMessageConsumer));
ASSERT_TRUE(IsEqual(env, variant_shader, context.get()));
}
TEST(TransformationInlineFunctionTest, OpPhiInBlockFollowingCall) {
// This test checks that if the block after the inlined function call has an
// OpPhi instruction and the called function contains multiple blocks then the
// OpPhi instruction gets updated to refer to the return block of the inlined
// function, since the block containing the call will no longer be a
// predecessor.
std::string reference_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
%13 = OpTypeBool
%14 = OpConstantTrue %13
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %10
%10 = OpLabel
%8 = OpFunctionCall %2 %6
OpBranch %11
%11 = OpLabel
%12 = OpPhi %13 %14 %10
OpReturn
OpFunctionEnd
%6 = OpFunction %2 None %3
%7 = OpLabel
OpBranch %20
%20 = OpLabel
OpReturn
OpFunctionEnd
)";
const auto env = SPV_ENV_UNIVERSAL_1_5;
const auto consumer = nullptr;
const auto context =
BuildModule(env, consumer, reference_shader, kFuzzAssembleOption);
spvtools::ValidatorOptions validator_options;
ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options,
kConsoleMessageConsumer));
TransformationContext transformation_context(
MakeUnique<FactManager>(context.get()), validator_options);
auto transformation = TransformationInlineFunction(8, {{7, 100}, {20, 101}});
ASSERT_TRUE(
transformation.IsApplicable(context.get(), transformation_context));
ApplyAndCheckFreshIds(transformation, context.get(), &transformation_context);
ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options,
kConsoleMessageConsumer));
std::string variant_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
%13 = OpTypeBool
%14 = OpConstantTrue %13
%4 = OpFunction %2 None %3
%5 = OpLabel
OpBranch %10
%10 = OpLabel
OpBranch %101
%101 = OpLabel
OpBranch %11
%11 = OpLabel
%12 = OpPhi %13 %14 %101
OpReturn
OpFunctionEnd
%6 = OpFunction %2 None %3
%7 = OpLabel
OpBranch %20
%20 = OpLabel
OpReturn
OpFunctionEnd
)";
ASSERT_TRUE(fuzzerutil::IsValidAndWellFormed(context.get(), validator_options,
kConsoleMessageConsumer));
ASSERT_TRUE(IsEqual(env, variant_shader, context.get()));
}
} // namespace
} // namespace fuzz
} // namespace spvtools