SPIRV-Tools/test/text_to_binary.control_flow_test.cpp
alan-baker f3cec93665
Support SPV_KHR_terminate_invocation (#3568)
Covers:
- assembler
- disassembler
- validator
- optimizer

Co-authored-by: David Neto <dneto@google.com>
2020-07-22 11:45:02 -04:00

428 lines
16 KiB
C++

// Copyright (c) 2015-2016 The Khronos Group Inc.
//
// 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.
// Assembler tests for instructions in the "Control Flow" section of the
// SPIR-V spec.
#include <sstream>
#include <string>
#include <tuple>
#include <vector>
#include "gmock/gmock.h"
#include "test/test_fixture.h"
#include "test/unit_spirv.h"
namespace spvtools {
namespace {
using spvtest::Concatenate;
using spvtest::EnumCase;
using spvtest::MakeInstruction;
using spvtest::TextToBinaryTest;
using ::testing::Combine;
using ::testing::Eq;
using ::testing::TestWithParam;
using ::testing::Values;
using ::testing::ValuesIn;
// Test OpSelectionMerge
using OpSelectionMergeTest = spvtest::TextToBinaryTestBase<
TestWithParam<EnumCase<SpvSelectionControlMask>>>;
TEST_P(OpSelectionMergeTest, AnySingleSelectionControlMask) {
const std::string input = "OpSelectionMerge %1 " + GetParam().name();
EXPECT_THAT(
CompiledInstructions(input),
Eq(MakeInstruction(SpvOpSelectionMerge, {1, GetParam().value()})));
}
// clang-format off
#define CASE(VALUE,NAME) { SpvSelectionControl##VALUE, NAME}
INSTANTIATE_TEST_SUITE_P(TextToBinarySelectionMerge, OpSelectionMergeTest,
ValuesIn(std::vector<EnumCase<SpvSelectionControlMask>>{
CASE(MaskNone, "None"),
CASE(FlattenMask, "Flatten"),
CASE(DontFlattenMask, "DontFlatten"),
}));
#undef CASE
// clang-format on
TEST_F(OpSelectionMergeTest, CombinedSelectionControlMask) {
const std::string input = "OpSelectionMerge %1 Flatten|DontFlatten";
const uint32_t expected_mask =
SpvSelectionControlFlattenMask | SpvSelectionControlDontFlattenMask;
EXPECT_THAT(CompiledInstructions(input),
Eq(MakeInstruction(SpvOpSelectionMerge, {1, expected_mask})));
}
TEST_F(OpSelectionMergeTest, WrongSelectionControl) {
// Case sensitive: "flatten" != "Flatten" and thus wrong.
EXPECT_THAT(CompileFailure("OpSelectionMerge %1 flatten|DontFlatten"),
Eq("Invalid selection control operand 'flatten|DontFlatten'."));
}
// Test OpLoopMerge
using OpLoopMergeTest = spvtest::TextToBinaryTestBase<
TestWithParam<std::tuple<spv_target_env, EnumCase<int>>>>;
TEST_P(OpLoopMergeTest, AnySingleLoopControlMask) {
const auto ctrl = std::get<1>(GetParam());
std::ostringstream input;
input << "OpLoopMerge %merge %continue " << ctrl.name();
for (auto num : ctrl.operands()) input << " " << num;
EXPECT_THAT(CompiledInstructions(input.str(), std::get<0>(GetParam())),
Eq(MakeInstruction(SpvOpLoopMerge, {1, 2, ctrl.value()},
ctrl.operands())));
}
#define CASE(VALUE, NAME) \
{ SpvLoopControl##VALUE, NAME }
#define CASE1(VALUE, NAME, PARM) \
{ \
SpvLoopControl##VALUE, NAME, { PARM } \
}
INSTANTIATE_TEST_SUITE_P(
TextToBinaryLoopMerge, OpLoopMergeTest,
Combine(Values(SPV_ENV_UNIVERSAL_1_0, SPV_ENV_UNIVERSAL_1_1),
ValuesIn(std::vector<EnumCase<int>>{
// clang-format off
CASE(MaskNone, "None"),
CASE(UnrollMask, "Unroll"),
CASE(DontUnrollMask, "DontUnroll"),
// clang-format on
})));
INSTANTIATE_TEST_SUITE_P(
TextToBinaryLoopMergeV11, OpLoopMergeTest,
Combine(Values(SPV_ENV_UNIVERSAL_1_1),
ValuesIn(std::vector<EnumCase<int>>{
// clang-format off
CASE(DependencyInfiniteMask, "DependencyInfinite"),
CASE1(DependencyLengthMask, "DependencyLength", 234),
{SpvLoopControlUnrollMask|SpvLoopControlDependencyLengthMask,
"DependencyLength|Unroll", {33}},
// clang-format on
})));
#undef CASE
#undef CASE1
TEST_F(OpLoopMergeTest, CombinedLoopControlMask) {
const std::string input = "OpLoopMerge %merge %continue Unroll|DontUnroll";
const uint32_t expected_mask =
SpvLoopControlUnrollMask | SpvLoopControlDontUnrollMask;
EXPECT_THAT(CompiledInstructions(input),
Eq(MakeInstruction(SpvOpLoopMerge, {1, 2, expected_mask})));
}
TEST_F(OpLoopMergeTest, WrongLoopControl) {
EXPECT_THAT(CompileFailure("OpLoopMerge %m %c none"),
Eq("Invalid loop control operand 'none'."));
}
// Test OpSwitch
TEST_F(TextToBinaryTest, SwitchGoodZeroTargets) {
EXPECT_THAT(CompiledInstructions("OpSwitch %selector %default"),
Eq(MakeInstruction(SpvOpSwitch, {1, 2})));
}
TEST_F(TextToBinaryTest, SwitchGoodOneTarget) {
EXPECT_THAT(CompiledInstructions("%1 = OpTypeInt 32 0\n"
"%2 = OpConstant %1 52\n"
"OpSwitch %2 %default 12 %target0"),
Eq(Concatenate({MakeInstruction(SpvOpTypeInt, {1, 32, 0}),
MakeInstruction(SpvOpConstant, {1, 2, 52}),
MakeInstruction(SpvOpSwitch, {2, 3, 12, 4})})));
}
TEST_F(TextToBinaryTest, SwitchGoodTwoTargets) {
EXPECT_THAT(
CompiledInstructions("%1 = OpTypeInt 32 0\n"
"%2 = OpConstant %1 52\n"
"OpSwitch %2 %default 12 %target0 42 %target1"),
Eq(Concatenate({
MakeInstruction(SpvOpTypeInt, {1, 32, 0}),
MakeInstruction(SpvOpConstant, {1, 2, 52}),
MakeInstruction(SpvOpSwitch, {2, 3, 12, 4, 42, 5}),
})));
}
TEST_F(TextToBinaryTest, SwitchBadMissingSelector) {
EXPECT_THAT(CompileFailure("OpSwitch"),
Eq("Expected operand, found end of stream."));
}
TEST_F(TextToBinaryTest, SwitchBadInvalidSelector) {
EXPECT_THAT(CompileFailure("OpSwitch 12"),
Eq("Expected id to start with %."));
}
TEST_F(TextToBinaryTest, SwitchBadMissingDefault) {
EXPECT_THAT(CompileFailure("OpSwitch %selector"),
Eq("Expected operand, found end of stream."));
}
TEST_F(TextToBinaryTest, SwitchBadInvalidDefault) {
EXPECT_THAT(CompileFailure("OpSwitch %selector 12"),
Eq("Expected id to start with %."));
}
TEST_F(TextToBinaryTest, SwitchBadInvalidLiteral) {
// The assembler recognizes "OpSwitch %selector %default" as a complete
// instruction. Then it tries to parse "%abc" as the start of a new
// instruction, but can't since it hits the end of stream.
const auto input = R"(%i32 = OpTypeInt 32 0
%selector = OpConstant %i32 42
OpSwitch %selector %default %abc)";
EXPECT_THAT(CompileFailure(input), Eq("Expected '=', found end of stream."));
}
TEST_F(TextToBinaryTest, SwitchBadMissingTarget) {
EXPECT_THAT(CompileFailure("%1 = OpTypeInt 32 0\n"
"%2 = OpConstant %1 52\n"
"OpSwitch %2 %default 12"),
Eq("Expected operand, found end of stream."));
}
// A test case for an OpSwitch.
// It is also parameterized to test encodings OpConstant
// integer literals. This can capture both single and multi-word
// integer literal tests.
struct SwitchTestCase {
std::string constant_type_args;
std::string constant_value_arg;
std::string case_value_arg;
std::vector<uint32_t> expected_instructions;
};
using OpSwitchValidTest =
spvtest::TextToBinaryTestBase<TestWithParam<SwitchTestCase>>;
// Tests the encoding of OpConstant literal values, and also
// the literal integer cases in an OpSwitch. This can
// test both single and multi-word integer literal encodings.
TEST_P(OpSwitchValidTest, ValidTypes) {
const std::string input = "%1 = OpTypeInt " + GetParam().constant_type_args +
"\n"
"%2 = OpConstant %1 " +
GetParam().constant_value_arg +
"\n"
"OpSwitch %2 %default " +
GetParam().case_value_arg + " %4\n";
std::vector<uint32_t> instructions;
EXPECT_THAT(CompiledInstructions(input),
Eq(GetParam().expected_instructions));
}
// Constructs a SwitchTestCase from the given integer_width, signedness,
// constant value string, and expected encoded constant.
SwitchTestCase MakeSwitchTestCase(uint32_t integer_width,
uint32_t integer_signedness,
std::string constant_str,
std::vector<uint32_t> encoded_constant,
std::string case_value_str,
std::vector<uint32_t> encoded_case_value) {
std::stringstream ss;
ss << integer_width << " " << integer_signedness;
return SwitchTestCase{
ss.str(),
constant_str,
case_value_str,
{Concatenate(
{MakeInstruction(SpvOpTypeInt,
{1, integer_width, integer_signedness}),
MakeInstruction(SpvOpConstant,
Concatenate({{1, 2}, encoded_constant})),
MakeInstruction(SpvOpSwitch,
Concatenate({{2, 3}, encoded_case_value, {4}}))})}};
}
INSTANTIATE_TEST_SUITE_P(
TextToBinaryOpSwitchValid1Word, OpSwitchValidTest,
ValuesIn(std::vector<SwitchTestCase>({
MakeSwitchTestCase(32, 0, "42", {42}, "100", {100}),
MakeSwitchTestCase(32, 1, "-1", {0xffffffff}, "100", {100}),
// SPIR-V 1.0 Rev 1 clarified that for an integer narrower than 32-bits,
// its bits will appear in the lower order bits of the 32-bit word, and
// a signed integer is sign-extended.
MakeSwitchTestCase(7, 0, "127", {127}, "100", {100}),
MakeSwitchTestCase(14, 0, "99", {99}, "100", {100}),
MakeSwitchTestCase(16, 0, "65535", {65535}, "100", {100}),
MakeSwitchTestCase(16, 1, "101", {101}, "100", {100}),
// Demonstrate sign extension
MakeSwitchTestCase(16, 1, "-2", {0xfffffffe}, "100", {100}),
// Hex cases
MakeSwitchTestCase(16, 1, "0x7ffe", {0x7ffe}, "0x1234", {0x1234}),
MakeSwitchTestCase(16, 1, "0x8000", {0xffff8000}, "0x8100",
{0xffff8100}),
MakeSwitchTestCase(16, 0, "0x8000", {0x00008000}, "0x8100", {0x8100}),
})));
// NB: The words LOW ORDER bits show up first.
INSTANTIATE_TEST_SUITE_P(
TextToBinaryOpSwitchValid2Words, OpSwitchValidTest,
ValuesIn(std::vector<SwitchTestCase>({
MakeSwitchTestCase(33, 0, "101", {101, 0}, "500", {500, 0}),
MakeSwitchTestCase(48, 1, "-1", {0xffffffff, 0xffffffff}, "900",
{900, 0}),
MakeSwitchTestCase(64, 1, "-2", {0xfffffffe, 0xffffffff}, "-5",
{0xfffffffb, uint32_t(-1)}),
// Hex cases
MakeSwitchTestCase(48, 1, "0x7fffffffffff", {0xffffffff, 0x00007fff},
"100", {100, 0}),
MakeSwitchTestCase(48, 1, "0x800000000000", {0x00000000, 0xffff8000},
"0x800000000000", {0x00000000, 0xffff8000}),
MakeSwitchTestCase(48, 0, "0x800000000000", {0x00000000, 0x00008000},
"0x800000000000", {0x00000000, 0x00008000}),
MakeSwitchTestCase(63, 0, "0x500000000", {0, 5}, "12", {12, 0}),
MakeSwitchTestCase(64, 0, "0x600000000", {0, 6}, "12", {12, 0}),
MakeSwitchTestCase(64, 1, "0x700000123", {0x123, 7}, "12", {12, 0}),
})));
using ControlFlowRoundTripTest = RoundTripTest;
TEST_P(ControlFlowRoundTripTest, DisassemblyEqualsAssemblyInput) {
const std::string assembly = GetParam();
EXPECT_THAT(EncodeAndDecodeSuccessfully(assembly), Eq(assembly)) << assembly;
}
INSTANTIATE_TEST_SUITE_P(
OpSwitchRoundTripUnsignedIntegers, ControlFlowRoundTripTest,
ValuesIn(std::vector<std::string>({
// Unsigned 16-bit.
"%1 = OpTypeInt 16 0\n%2 = OpConstant %1 65535\nOpSwitch %2 %3\n",
// Unsigned 32-bit, three non-default cases.
"%1 = OpTypeInt 32 0\n%2 = OpConstant %1 123456\n"
"OpSwitch %2 %3 100 %4 102 %5 1000000 %6\n",
// Unsigned 48-bit, three non-default cases.
"%1 = OpTypeInt 48 0\n%2 = OpConstant %1 5000000000\n"
"OpSwitch %2 %3 100 %4 102 %5 6000000000 %6\n",
// Unsigned 64-bit, three non-default cases.
"%1 = OpTypeInt 64 0\n%2 = OpConstant %1 9223372036854775807\n"
"OpSwitch %2 %3 100 %4 102 %5 9000000000000000000 %6\n",
})));
INSTANTIATE_TEST_SUITE_P(
OpSwitchRoundTripSignedIntegers, ControlFlowRoundTripTest,
ValuesIn(std::vector<std::string>{
// Signed 16-bit, with two non-default cases
"%1 = OpTypeInt 16 1\n%2 = OpConstant %1 32767\n"
"OpSwitch %2 %3 99 %4 -102 %5\n",
"%1 = OpTypeInt 16 1\n%2 = OpConstant %1 -32768\n"
"OpSwitch %2 %3 99 %4 -102 %5\n",
// Signed 32-bit, two non-default cases.
"%1 = OpTypeInt 32 1\n%2 = OpConstant %1 -123456\n"
"OpSwitch %2 %3 100 %4 -123456 %5\n",
"%1 = OpTypeInt 32 1\n%2 = OpConstant %1 123456\n"
"OpSwitch %2 %3 100 %4 123456 %5\n",
// Signed 48-bit, three non-default cases.
"%1 = OpTypeInt 48 1\n%2 = OpConstant %1 5000000000\n"
"OpSwitch %2 %3 100 %4 -7000000000 %5 6000000000 %6\n",
"%1 = OpTypeInt 48 1\n%2 = OpConstant %1 -5000000000\n"
"OpSwitch %2 %3 100 %4 -7000000000 %5 6000000000 %6\n",
// Signed 64-bit, three non-default cases.
"%1 = OpTypeInt 64 1\n%2 = OpConstant %1 9223372036854775807\n"
"OpSwitch %2 %3 100 %4 7000000000 %5 -1000000000000000000 %6\n",
"%1 = OpTypeInt 64 1\n%2 = OpConstant %1 -9223372036854775808\n"
"OpSwitch %2 %3 100 %4 7000000000 %5 -1000000000000000000 %6\n",
}));
using OpSwitchInvalidTypeTestCase =
spvtest::TextToBinaryTestBase<TestWithParam<std::string>>;
TEST_P(OpSwitchInvalidTypeTestCase, InvalidTypes) {
const std::string input =
"%1 = " + GetParam() +
"\n"
"%3 = OpCopyObject %1 %2\n" // We only care the type of the expression
" OpSwitch %3 %default 32 %c\n";
EXPECT_THAT(CompileFailure(input),
Eq("The selector operand for OpSwitch must be the result of an "
"instruction that generates an integer scalar"));
}
// clang-format off
INSTANTIATE_TEST_SUITE_P(
TextToBinaryOpSwitchInvalidTests, OpSwitchInvalidTypeTestCase,
ValuesIn(std::vector<std::string>{
{"OpTypeVoid",
"OpTypeBool",
"OpTypeFloat 32",
"OpTypeVector %a 32",
"OpTypeMatrix %a 32",
"OpTypeImage %a 1D 0 0 0 0 Unknown",
"OpTypeSampler",
"OpTypeSampledImage %a",
"OpTypeArray %a %b",
"OpTypeRuntimeArray %a",
"OpTypeStruct %a",
"OpTypeOpaque \"Foo\"",
"OpTypePointer UniformConstant %a",
"OpTypeFunction %a %b",
"OpTypeEvent",
"OpTypeDeviceEvent",
"OpTypeReserveId",
"OpTypeQueue",
"OpTypePipe ReadOnly",
// Skip OpTypeForwardPointer becasuse it doesn't even produce a result
// ID.
// At least one thing that isn't a type at all
"OpNot %a %b"
},
}));
// clang-format on
using OpKillTest = spvtest::TextToBinaryTest;
INSTANTIATE_TEST_SUITE_P(OpKillTest, ControlFlowRoundTripTest,
Values("OpKill\n"));
TEST_F(OpKillTest, ExtraArgsAssemblyError) {
const std::string input = "OpKill 1";
EXPECT_THAT(CompileFailure(input),
Eq("Expected <opcode> or <result-id> at the beginning of an "
"instruction, found '1'."));
}
using OpTerminateInvocationTest = spvtest::TextToBinaryTest;
INSTANTIATE_TEST_SUITE_P(OpTerminateInvocationTest, ControlFlowRoundTripTest,
Values("OpTerminateInvocation\n"));
TEST_F(OpTerminateInvocationTest, ExtraArgsAssemblyError) {
const std::string input = "OpTerminateInvocation 1";
EXPECT_THAT(CompileFailure(input),
Eq("Expected <opcode> or <result-id> at the beginning of an "
"instruction, found '1'."));
}
// TODO(dneto): OpPhi
// TODO(dneto): OpLoopMerge
// TODO(dneto): OpLabel
// TODO(dneto): OpBranch
// TODO(dneto): OpSwitch
// TODO(dneto): OpReturn
// TODO(dneto): OpReturnValue
// TODO(dneto): OpUnreachable
// TODO(dneto): OpLifetimeStart
// TODO(dneto): OpLifetimeStop
} // namespace
} // namespace spvtools