mirror of
https://github.com/KhronosGroup/SPIRV-Tools
synced 2024-11-22 19:50:05 +00:00
7e1825a596
Fixes #3938.
1117 lines
39 KiB
C++
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
|