mirror of
https://github.com/zebrajr/pytorch.git
synced 2026-01-15 12:15:51 +00:00
8fa81a6066
Enables clang-tidy rule [`misc-use-internal-linkage`](https://clang.llvm.org/extra/clang-tidy/checks/misc/use-internal-linkage.html). This new check was introduced in Clang-Tidy 18 and is available due to recent update of Clang-Tidy 19. The check marks functions and variables used only in the translation unit as static. Therefore undesired symbols are not leaked into other units, more link time optimisations are possible and the resulting binaries may be smaller. The detected violations were mostly fixed by using static. In other cases, the symbols were indeed consumed by others files, then their declaring headers were included. Still some declarations were wrong and have been fixed. Pull Request resolved: https://github.com/pytorch/pytorch/pull/148948 Approved by: https://github.com/Skylion007
89 lines
2.1 KiB
C++
89 lines
2.1 KiB
C++
// Copyright 2004-present Facebook. All Rights Reserved.
|
|
|
|
#include <c10/util/irange.h>
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <array>
|
|
|
|
using namespace ::testing;
|
|
|
|
TEST(irangeTest, range_test) {
|
|
std::vector<int> test_vec;
|
|
for (const auto i : c10::irange(4, 11)) {
|
|
test_vec.push_back(i);
|
|
}
|
|
const std::vector<int> correct = {{4, 5, 6, 7, 8, 9, 10}};
|
|
ASSERT_EQ(test_vec, correct);
|
|
}
|
|
|
|
TEST(irangeTest, end_test) {
|
|
std::vector<int> test_vec;
|
|
for (const auto i : c10::irange(5)) {
|
|
test_vec.push_back(i);
|
|
}
|
|
const std::vector<int> correct = {{0, 1, 2, 3, 4}};
|
|
ASSERT_EQ(test_vec, correct);
|
|
}
|
|
|
|
TEST(irangeTest, neg_range_test) {
|
|
std::vector<int> test_vec;
|
|
for (const auto i : c10::irange(-2, 3)) {
|
|
test_vec.push_back(i);
|
|
}
|
|
const std::vector<int> correct = {{-2, -1, 0, 1, 2}};
|
|
ASSERT_EQ(test_vec, correct);
|
|
}
|
|
|
|
TEST(irange, empty_reverse_range_two_inputs) {
|
|
std::vector<int> test_vec;
|
|
for (const auto i : c10::irange(3, -3)) {
|
|
test_vec.push_back(i);
|
|
if (i > 20) { // Cap the number of elements we add if something goes wrong
|
|
break;
|
|
}
|
|
}
|
|
const std::vector<int> correct = {};
|
|
ASSERT_EQ(test_vec, correct);
|
|
}
|
|
|
|
TEST(irange, empty_reverse_range_one_input) {
|
|
std::vector<int> test_vec;
|
|
for (const auto i : c10::irange(-3)) {
|
|
test_vec.push_back(i);
|
|
if (i > 20) { // Cap the number of elements we add if something goes wrong
|
|
break;
|
|
}
|
|
}
|
|
const std::vector<int> correct = {};
|
|
ASSERT_EQ(test_vec, correct);
|
|
}
|
|
|
|
static constexpr std::array<int, 3> toy_iota() {
|
|
std::array<int, 3> result = {0};
|
|
for (const auto i : c10::irange(3)) {
|
|
result[i] = i;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
static constexpr std::array<int, 3> toy_iota_with_start(int start) {
|
|
std::array<int, 3> result = {0};
|
|
for (const auto i : c10::irange(start, start + 3)) {
|
|
result[i - start] = i;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
TEST(irange, constexpr_ok) {
|
|
constexpr auto arr = toy_iota();
|
|
static_assert(arr[0] == 0);
|
|
static_assert(arr[1] == 1);
|
|
static_assert(arr[2] == 2);
|
|
|
|
constexpr auto arr2 = toy_iota_with_start(4);
|
|
static_assert(arr2[0] == 4);
|
|
static_assert(arr2[1] == 5);
|
|
static_assert(arr2[2] == 6);
|
|
}
|