mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
0c0d8c8ff0
Before we can use clang-tidy-17 Pull Request resolved: https://github.com/pytorch/pytorch/pull/137407 Approved by: https://github.com/Skylion007, https://github.com/aaronenyeshi
23 lines
557 B
C++
23 lines
557 B
C++
#include <gtest/gtest.h>
|
|
|
|
#include <iostream>
|
|
|
|
#include <c10/util/Flags.h>
|
|
|
|
C10_DEFINE_bool(c10_flags_test_only_flag, true, "Only used in test.");
|
|
|
|
namespace c10_test {
|
|
|
|
TEST(FlagsTest, TestGflagsCorrectness) {
|
|
#ifdef C10_USE_GFLAGS
|
|
EXPECT_EQ(FLAGS_c10_flags_test_only_flag, true);
|
|
FLAGS_c10_flags_test_only_flag = false;
|
|
FLAGS_c10_flags_test_only_flag = true;
|
|
EXPECT_EQ(FLAGS_c10_flags_test_only_flag, true);
|
|
#else // C10_USE_GFLAGS
|
|
std::cout << "Caffe2 is not built with gflags. Nothing to test here." << '\n';
|
|
#endif
|
|
}
|
|
|
|
} // namespace c10_test
|