mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
a9b0a921d5
Summary: As GoogleTest `TEST` macro is non-compliant with it as well as `DEFINE_DISPATCH` All changes but the ones to `.clang-tidy` are generated using following script: ``` for i in `find . -type f -iname "*.c*" -or -iname "*.h"|xargs grep cppcoreguidelines-avoid-non-const-global-variables|cut -f1 -d:|sort|uniq`; do sed -i "/\/\/ NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)/d" $i; done ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/62008 Reviewed By: driazati, r-barnes Differential Revision: D29838584 Pulled By: malfet fbshipit-source-id: 1b2f8602c945bd4ce50a9bfdd204755556e31d13
38 lines
764 B
C++
38 lines
764 B
C++
#include "caffe2/operators/elementwise_op_test.h"
|
|
|
|
#include "caffe2/core/flags.h"
|
|
|
|
C10_DECLARE_string(caffe_test_root);
|
|
|
|
template <>
|
|
void CopyVector<caffe2::CPUContext, bool>(const int N, const bool* x, bool* y) {
|
|
memcpy(y, x, N * sizeof(bool));
|
|
}
|
|
|
|
template <>
|
|
void CopyVector<caffe2::CPUContext, int32_t>(
|
|
const int N,
|
|
const int32_t* x,
|
|
int32_t* y) {
|
|
memcpy(y, x, N * sizeof(int32_t));
|
|
}
|
|
|
|
TEST(ElementwiseCPUTest, And) {
|
|
elementwiseAnd<caffe2::CPUContext>();
|
|
}
|
|
|
|
TEST(ElementwiseTest, Or) {
|
|
elementwiseOr<caffe2::CPUContext>();
|
|
}
|
|
|
|
TEST(ElementwiseTest, Xor) {
|
|
elementwiseXor<caffe2::CPUContext>();
|
|
}
|
|
|
|
TEST(ElementwiseTest, Not) {
|
|
elementwiseNot<caffe2::CPUContext>();
|
|
}
|
|
|
|
TEST(ElementwiseTest, EQ) {
|
|
elementwiseEQ<caffe2::CPUContext>();
|
|
}
|