| // 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 |