mirror of
https://github.com/KhronosGroup/SPIRV-Tools
synced 2024-12-28 02:31:04 +00:00
d35a78db57
Fixes #4960 * Switches to using enum classes with an underlying type to avoid undefined behaviour
137 lines
4.7 KiB
C++
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
|