Function TEST

test/cpp/phi/api/test_from_blob.cc:42–79

}  // namespace paddle

TEST(from_blob, CPU) {
  // 1. create data
  int64_t data[] = {4, 3, 2, 1};  // NOLINT

  ASSERT_EQ(paddle::GetPlaceFromPtr(data), phi::CPUPlace());

  // 2. test API
  auto test_tensor = from_blob(data, {1, 2, 2}, DataType::INT64);

  // 3. check result
  // 3.1 check tensor attributes
  ASSERT_EQ(test_tensor.dims().size(), 3);
  ASSERT_EQ(test_tensor.dims()[0], 1);
  ASSERT_EQ(test_tensor.dims()[1], 2);
  ASSERT_EQ(test_tensor.dims()[2], 2);
  ASSERT_EQ(test_tensor.numel(), 4);
  ASSERT_EQ(test_tensor.is_cpu(), true);
  ASSERT_EQ(test_tensor.dtype(), DataType::INT64);
  ASSERT_EQ(test_tensor.layout(), phi::DataLayout::NCHW);
  ASSERT_EQ(test_tensor.is_dense_tensor(), true);

  // 3.2 check tensor values
  auto* test_tensor_data = test_tensor.template data<int64_t>();
  for (int64_t i = 0; i < 4; i++) {
    ASSERT_EQ(test_tensor_data[i], 4 - i);
  }

  // 3.3 check whether memory is shared
  ASSERT_EQ(data, test_tensor_data);

  // 3.4 test other API
  auto test_tensor_pow = paddle::experimental::pow(test_tensor, 2);
  auto* test_tensor_pow_data = test_tensor_pow.template data<int64_t>();
  for (int64_t i = 0; i < 4; i++) {
    ASSERT_EQ(test_tensor_pow_data[i],
              static_cast<int64_t>(std::pow(4 - i, 2)));
  }
}

#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)