mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08: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
1.1 KiB
C++
38 lines
1.1 KiB
C++
#include <random>
|
|
|
|
#include <gtest/gtest.h>
|
|
#include "caffe2/core/context.h"
|
|
#include "caffe2/proto/caffe2_pb.h"
|
|
|
|
namespace caffe2 {
|
|
|
|
TEST(CPUContextTest, TestAllocAlignment) {
|
|
for (int i = 1; i < 10; ++i) {
|
|
auto data = CPUContext::New(i);
|
|
EXPECT_EQ((reinterpret_cast<size_t>(data.get()) % gAlignment), 0);
|
|
// data is freed when out of scope
|
|
}
|
|
}
|
|
|
|
TEST(CPUContextTest, TestAllocDealloc) {
|
|
auto data_ptr = CPUContext::New(10 * sizeof(float));
|
|
float* data = static_cast<float*>(data_ptr.get());
|
|
EXPECT_NE(data, nullptr);
|
|
auto dst_data_ptr = CPUContext::New(10 * sizeof(float));
|
|
float* dst_data = static_cast<float*>(dst_data_ptr.get());
|
|
EXPECT_NE(dst_data, nullptr);
|
|
for (int i = 0; i < 10; ++i) {
|
|
// NOLINTNEXTLINE(cppcoreguidelines-narrowing-conversions,bugprone-narrowing-conversions)
|
|
data[i] = i;
|
|
}
|
|
DeviceOption option;
|
|
CPUContext context(option);
|
|
context.CopyToCPU<float>(10, data, dst_data);
|
|
for (int i = 0; i < 10; ++i) {
|
|
EXPECT_FLOAT_EQ(dst_data[i], i);
|
|
}
|
|
// data_ptr is freed when out of scope
|
|
}
|
|
|
|
} // namespace caffe2
|