mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01: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
46 lines
1.2 KiB
C++
46 lines
1.2 KiB
C++
#include <iostream>
|
|
|
|
#include "caffe2/core/flags.h"
|
|
#include "caffe2/operators/utility_ops.h"
|
|
#include <gtest/gtest.h>
|
|
|
|
C10_DECLARE_string(caffe_test_root);
|
|
|
|
namespace caffe2 {
|
|
|
|
static void AddConstInput(
|
|
const vector<int64_t>& shape,
|
|
const float value,
|
|
const string& name,
|
|
Workspace* ws) {
|
|
DeviceOption option;
|
|
CPUContext context(option);
|
|
Blob* blob = ws->CreateBlob(name);
|
|
auto* tensor = BlobGetMutableTensor(blob, CPU);
|
|
tensor->Resize(shape);
|
|
math::Set<float, CPUContext>(
|
|
tensor->numel(), value, tensor->template mutable_data<float>(), &context);
|
|
return;
|
|
}
|
|
|
|
TEST(UtilityOpTest, testReshapeWithScalar) {
|
|
Workspace ws;
|
|
OperatorDef def;
|
|
def.set_name("test_reshape");
|
|
def.set_type("Reshape");
|
|
def.add_input("X");
|
|
def.add_output("XNew");
|
|
def.add_output("OldShape");
|
|
def.add_arg()->CopyFrom(MakeArgument("shape", vector<int64_t>{1}));
|
|
AddConstInput(vector<int64_t>(), 3.14, "X", &ws);
|
|
// execute the op
|
|
unique_ptr<OperatorBase> op(CreateOperator(def, &ws));
|
|
EXPECT_TRUE(op->Run());
|
|
Blob* XNew = ws.GetBlob("XNew");
|
|
const TensorCPU& XNewTensor = XNew->Get<Tensor>();
|
|
EXPECT_EQ(1, XNewTensor.dim());
|
|
EXPECT_EQ(1, XNewTensor.numel());
|
|
}
|
|
|
|
} // namespace caffe2
|