SPIRV-Tools/test/link/global_values_amount_test.cpp
alan-baker d35a78db57
Switch SPIRV-Tools to use spirv.hpp11 internally (#4981)
Fixes #4960

* Switches to using enum classes with an underlying type to avoid
  undefined behaviour
2022-11-04 17:27:10 -04:00

137 lines
4.7 KiB
C++

// Copyright (c) 2017 Pierre Moreau
//
// 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 <string>
#include "gmock/gmock.h"
#include "test/link/linker_fixture.h"
namespace spvtools {
namespace {
using ::testing::HasSubstr;
const uint32_t binary_count = 2u;
class EntryPointsAmountTest : public spvtest::LinkerTest {
public:
EntryPointsAmountTest() { binaries.reserve(binary_count + 1u); }
void SetUp() override {
const uint32_t global_variable_count_per_binary =
(SPV_LIMIT_GLOBAL_VARIABLES_MAX - 1u) / binary_count;
spvtest::Binary common_binary = {
// clang-format off
static_cast<uint32_t>(spv::MagicNumber),
static_cast<uint32_t>(spv::Version),
SPV_GENERATOR_WORD(SPV_GENERATOR_KHRONOS, 0),
3u + global_variable_count_per_binary, // NOTE: Bound
0u, // NOTE: Schema; reserved
static_cast<uint32_t>(spv::Op::OpCapability) | 2u << spv::WordCountShift,
static_cast<uint32_t>(spv::Capability::Shader),
static_cast<uint32_t>(spv::Op::OpMemoryModel) | 3u << spv::WordCountShift,
static_cast<uint32_t>(spv::AddressingModel::Logical),
static_cast<uint32_t>(spv::MemoryModel::Simple),
static_cast<uint32_t>(spv::Op::OpTypeFloat) | 3u << spv::WordCountShift,
1u, // NOTE: Result ID
32u, // NOTE: Width
static_cast<uint32_t>(spv::Op::OpTypePointer) | 4u << spv::WordCountShift,
2u, // NOTE: Result ID
static_cast<uint32_t>(spv::StorageClass::Input),
1u // NOTE: Type ID
// clang-format on
};
binaries.push_back({});
spvtest::Binary& binary = binaries.back();
binary.reserve(common_binary.size() + global_variable_count_per_binary * 4);
binary.insert(binary.end(), common_binary.cbegin(), common_binary.cend());
for (uint32_t i = 0u; i < global_variable_count_per_binary; ++i) {
binary.push_back(static_cast<uint32_t>(spv::Op::OpVariable) |
4u << spv::WordCountShift);
binary.push_back(2u); // NOTE: Type ID
binary.push_back(3u + i); // NOTE: Result ID
binary.push_back(static_cast<uint32_t>(spv::StorageClass::Input));
}
for (uint32_t i = 0u; i < binary_count - 1u; ++i) {
binaries.push_back(binaries.back());
}
}
void TearDown() override { binaries.clear(); }
spvtest::Binaries binaries;
};
TEST_F(EntryPointsAmountTest, UnderLimit) {
spvtest::Binary linked_binary;
ASSERT_EQ(SPV_SUCCESS, Link(binaries, &linked_binary)) << GetErrorMessage();
EXPECT_THAT(GetErrorMessage(), std::string());
}
TEST_F(EntryPointsAmountTest, OverLimit) {
binaries.push_back({
// clang-format off
static_cast<uint32_t>(spv::MagicNumber),
static_cast<uint32_t>(spv::Version),
SPV_GENERATOR_WORD(SPV_GENERATOR_KHRONOS, 0),
5u, // NOTE: Bound
0u, // NOTE: Schema; reserved
static_cast<uint32_t>(spv::Op::OpCapability) | 2u << spv::WordCountShift,
static_cast<uint32_t>(spv::Capability::Shader),
static_cast<uint32_t>(spv::Op::OpMemoryModel) | 3u << spv::WordCountShift,
static_cast<uint32_t>(spv::AddressingModel::Logical),
static_cast<uint32_t>(spv::MemoryModel::Simple),
static_cast<uint32_t>(spv::Op::OpTypeFloat) | 3u << spv::WordCountShift,
1u, // NOTE: Result ID
32u, // NOTE: Width
static_cast<uint32_t>(spv::Op::OpTypePointer) | 4u << spv::WordCountShift,
2u, // NOTE: Result ID
static_cast<uint32_t>(spv::StorageClass::Input),
1u, // NOTE: Type ID
static_cast<uint32_t>(spv::Op::OpVariable) | 4u << spv::WordCountShift,
2u, // NOTE: Type ID
3u, // NOTE: Result ID
static_cast<uint32_t>(spv::StorageClass::Input),
static_cast<uint32_t>(spv::Op::OpVariable) | 4u << spv::WordCountShift,
2u, // NOTE: Type ID
4u, // NOTE: Result ID
static_cast<uint32_t>(spv::StorageClass::Input)
// clang-format on
});
spvtest::Binary linked_binary;
ASSERT_EQ(SPV_SUCCESS, Link(binaries, &linked_binary)) << GetErrorMessage();
EXPECT_THAT(
GetErrorMessage(),
HasSubstr("The minimum limit of global values, 65535, was exceeded; "
"65536 global values were found."));
}
} // namespace
} // namespace spvtools