mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
802d21c8f4
Summary: Continuing pjh5's work to remove FULL_CAFFE2 flag completely. With these changes you'll be able to also do something like ``` NO_TEST=1 python setup.py build_deps ``` and this will skip building tests in caffe2, aten, and c10d. By default the tests are built. cc mingzhe09088 Yangqing Pull Request resolved: https://github.com/pytorch/pytorch/pull/11321 Reviewed By: mingzhe09088 Differential Revision: D9694950 Pulled By: orionr fbshipit-source-id: ff5c4937a23d1a263378a196a5eda0cba98af0a8
95 lines
4.5 KiB
C++
95 lines
4.5 KiB
C++
#include "caffe2/utils/Array.h"
|
|
#include <gtest/gtest.h>
|
|
|
|
using c10::guts::array;
|
|
using c10::guts::to_array;
|
|
|
|
namespace {
|
|
namespace test_equals {
|
|
static_assert(array<int, 0>{{}} == array<int, 0>{{}}, "");
|
|
static_assert(array<int, 3>{{2, 3, 4}} == array<int, 3>{{2, 3, 4}}, "");
|
|
static_assert(!(array<int, 3>{{2, 3, 4}} == array<int, 3>{{1, 3, 4}}), "");
|
|
static_assert(!(array<int, 3>{{2, 3, 4}} == array<int, 3>{{2, 1, 4}}), "");
|
|
static_assert(!(array<int, 3>{{2, 3, 4}} == array<int, 3>{{2, 3, 1}}), "");
|
|
}
|
|
|
|
namespace test_notequals {
|
|
static_assert(!(array<int, 0>{{}} != array<int, 0>{{}}), "");
|
|
static_assert(!(array<int, 3>{{2, 3, 4}} != array<int, 3>{{2, 3, 4}}), "");
|
|
static_assert(array<int, 3>{{2, 3, 4}} != array<int, 3>{{1, 3, 4}}, "");
|
|
static_assert(array<int, 3>{{2, 3, 4}} != array<int, 3>{{2, 1, 4}}, "");
|
|
static_assert(array<int, 3>{{2, 3, 4}} != array<int, 3>{{2, 3, 1}}, "");
|
|
}
|
|
|
|
namespace test_lessthan {
|
|
static_assert(!(array<int, 0>{{}} < array<int, 0>{{}}), "");
|
|
static_assert(!(array<int, 1>{{2}} < array<int, 1>{{1}}), "");
|
|
static_assert(array<int, 1>{{1}} < array<int, 1>{{2}}, "");
|
|
static_assert(!(array<int, 3>{{1, 2, 3}} < array<int, 3>{{1, 2, 3}}), "");
|
|
static_assert(array<int, 3>{{1, 2, 3}} < array<int, 3>{{2, 2, 3}}, "");
|
|
static_assert(!(array<int, 3>{{1, 2, 3}} < array<int, 3>{{0, 2, 3}}), "");
|
|
static_assert(array<int, 3>{{1, 2, 3}} < array<int, 3>{{1, 3, 3}}, "");
|
|
static_assert(!(array<int, 3>{{1, 2, 3}} < array<int, 3>{{1, 1, 3}}), "");
|
|
static_assert(array<int, 3>{{1, 2, 3}} < array<int, 3>{{1, 2, 4}}, "");
|
|
static_assert(!(array<int, 3>{{1, 2, 3}} < array<int, 3>{{1, 2, 2}}), "");
|
|
}
|
|
|
|
namespace test_greaterthan {
|
|
static_assert(!(array<int, 0>{{}} > array<int, 0>{{}}), "");
|
|
static_assert(!(array<int, 1>{{1}} > array<int, 1>{{2}}), "");
|
|
static_assert(array<int, 1>{{2}} > array<int, 1>{{1}}, "");
|
|
static_assert(!(array<int, 3>{{1, 2, 3}} > array<int, 3>{{1, 2, 3}}), "");
|
|
static_assert(array<int, 3>{{2, 2, 3}} > array<int, 3>{{1, 2, 3}}, "");
|
|
static_assert(!(array<int, 3>{{0, 2, 3}} > array<int, 3>{{1, 2, 3}}), "");
|
|
static_assert(array<int, 3>{{1, 3, 3}} > array<int, 3>{{1, 2, 3}}, "");
|
|
static_assert(!(array<int, 3>{{1, 1, 3}} > array<int, 3>{{1, 2, 3}}), "");
|
|
static_assert(array<int, 3>{{1, 2, 4}} > array<int, 3>{{1, 2, 3}}, "");
|
|
static_assert(!(array<int, 3>{{1, 2, 2}} > array<int, 3>{{1, 2, 3}}), "");
|
|
}
|
|
|
|
namespace test_lessequals {
|
|
static_assert(array<int, 0>{{}} <= array<int, 0>{{}}, "");
|
|
static_assert(!(array<int, 1>{{2}} <= array<int, 1>{{1}}), "");
|
|
static_assert(array<int, 1>{{1}} <= array<int, 1>{{2}}, "");
|
|
static_assert(array<int, 3>{{1, 2, 3}} <= array<int, 3>{{1, 2, 3}}, "");
|
|
static_assert(array<int, 3>{{1, 2, 3}} <= array<int, 3>{{2, 2, 3}}, "");
|
|
static_assert(!(array<int, 3>{{1, 2, 3}} <= array<int, 3>{{0, 2, 3}}), "");
|
|
static_assert(array<int, 3>{{1, 2, 3}} <= array<int, 3>{{1, 3, 3}}, "");
|
|
static_assert(!(array<int, 3>{{1, 2, 3}} <= array<int, 3>{{1, 1, 3}}), "");
|
|
static_assert(array<int, 3>{{1, 2, 3}} <= array<int, 3>{{1, 2, 4}}, "");
|
|
static_assert(!(array<int, 3>{{1, 2, 3}} <= array<int, 3>{{1, 2, 2}}), "");
|
|
}
|
|
|
|
namespace test_greaterequals {
|
|
static_assert(array<int, 0>{{}} >= array<int, 0>{{}}, "");
|
|
static_assert(!(array<int, 1>{{1}} >= array<int, 1>{{2}}), "");
|
|
static_assert(array<int, 1>{{2}} >= array<int, 1>{{1}}, "");
|
|
static_assert(array<int, 3>{{1, 2, 3}} >= array<int, 3>{{1, 2, 3}}, "");
|
|
static_assert(array<int, 3>{{2, 2, 3}} >= array<int, 3>{{1, 2, 3}}, "");
|
|
static_assert(!(array<int, 3>{{0, 2, 3}} >= array<int, 3>{{1, 2, 3}}), "");
|
|
static_assert(array<int, 3>{{1, 3, 3}} >= array<int, 3>{{1, 2, 3}}, "");
|
|
static_assert(!(array<int, 3>{{1, 1, 3}} >= array<int, 3>{{1, 2, 3}}), "");
|
|
static_assert(array<int, 3>{{1, 2, 4}} >= array<int, 3>{{1, 2, 3}}, "");
|
|
static_assert(!(array<int, 3>{{1, 2, 2}} >= array<int, 3>{{1, 2, 3}}), "");
|
|
}
|
|
|
|
namespace test_tail {
|
|
static_assert(array < int, 2 > {{3, 4}} == tail(array < int, 3 > {{2, 3, 4}}), "");
|
|
static_assert(array < int, 0 > {{}} == tail(array < int, 1 > {{3}}), "");
|
|
}
|
|
|
|
TEST(ArrayTest, TestPrepend) {
|
|
// Some compilers can't handle move results as constexpr, so use
|
|
// gtest assert for this test
|
|
ASSERT_EQ((array<int, 3> {{2, 3, 4}}), (prepend(2, array<int, 2> {{3, 4}})));
|
|
ASSERT_EQ((array<int, 1> {{3}}), (prepend(3, array<int, 0> {{}})));
|
|
}
|
|
|
|
namespace test_to_std_array {
|
|
constexpr int obj2[3] = {3, 5, 6};
|
|
static_assert(array < int, 3 > {{3, 5, 6}} == to_array(obj2), "");
|
|
static_assert(array < int, 3 > {{3, 5, 6}} == to_array<int, 3>({3, 5, 6}), "");
|
|
}
|
|
|
|
}
|