vol2col_test.cc 4.0 KB
Newer Older
C
chengduoZH 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
/* Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserve.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License. */

#include "paddle/operators/math/vol2col.h"
#include <gtest/gtest.h>
#include <iostream>

template <typename Place>
void testVol2col() {
  paddle::framework::Tensor input;
C
chengduoZH 已提交
22 23
  paddle::framework::Tensor input_tmp;
  paddle::framework::Tensor output;
C
chengduoZH 已提交
24 25 26 27 28 29 30 31
  paddle::framework::Tensor output_tmp;

  auto* place = new Place();
  paddle::platform::DeviceContext* context;
  if (paddle::platform::is_cpu_place(*place)) {
    context =
        new paddle::platform::CPUDeviceContext(paddle::platform::CPUPlace());
  } else {
C
chengduoZH 已提交
32
#ifdef PADDLE_WITH_CUDA
C
chengduoZH 已提交
33 34 35 36
    context =
        new paddle::platform::CUDADeviceContext(paddle::platform::GPUPlace());
#else
    PADDLE_THROW("no GPU support");
C
chengduoZH 已提交
37
#endif  // PADDLE_WITH_CUDA
C
chengduoZH 已提交
38 39 40 41 42 43 44 45
  }

  /**
   * input = [[0, 1, 2,
   *          3, 4, 5]
   *          [6, 7, 8,
   *          9, 10, 11]]
   *
C
chengduoZH 已提交
46 47 48 49 50 51 52 53
   * output = [0, 1
   *           1, 2
   *           3, 4
   *           4, 5
   *           6, 7
   *           7, 8
   *           9, 10
   *           10, 11]
C
chengduoZH 已提交
54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82
   *
   * col2vol = [[0, 2, 2,
   *             3, 8, 5]
   *            [6, 14, 8,
   *             9, 20, 11]]
   *
   */
  int input_depth = 2;
  int input_height = 2;
  int input_width = 3;
  int filter_size = 2;
  int stride = 1;
  int padding = 0;
  int output_depth = (input_depth - filter_size + 2 * padding) / stride + 1;
  int output_height = (input_height - filter_size + 2 * padding) / stride + 1;
  int output_width = (input_width - filter_size + 2 * padding) / stride + 1;

  // Vol2Col test
  float* input_ptr =
      input_tmp.mutable_data<float>({1, input_depth, input_height, input_width},
                                    paddle::platform::CPUPlace());
  float arr[12] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11};
  memcpy(input_ptr, arr, 12 * sizeof(float));

  if (paddle::platform::is_cpu_place(*place)) {
    input = input_tmp;
  } else {
    input.CopyFrom<float>(input_tmp, *place);
  }
C
chengduoZH 已提交
83 84 85
  output.mutable_data<float>({1, filter_size, filter_size, filter_size,
                              output_depth, output_height, output_width},
                             *place);
C
chengduoZH 已提交
86 87

  paddle::operators::math::Vol2ColFunctor<Place, float> vol2col;
C
chengduoZH 已提交
88
  vol2col(*context, input, output, stride, stride, stride, padding, padding,
C
chengduoZH 已提交
89 90
          padding);

C
chengduoZH 已提交
91
  float vol_2_col[] = {0, 1, 1, 2, 3, 4, 4, 5, 6, 7, 7, 8, 9, 10, 10, 11};
C
chengduoZH 已提交
92 93
  float* out_cfo_ptr;
  if (paddle::platform::is_cpu_place(*place)) {
C
chengduoZH 已提交
94
    out_cfo_ptr = output.data<float>();
C
chengduoZH 已提交
95
  } else {
C
chengduoZH 已提交
96
    output_tmp.CopyFrom<float>(output, paddle::platform::CPUPlace());
C
chengduoZH 已提交
97 98 99
    out_cfo_ptr = output_tmp.data<float>();
  }

C
chengduoZH 已提交
100 101 102
  for (int i = 0; i < 16; ++i) {
    EXPECT_EQ(out_cfo_ptr[i], vol_2_col[i]);
  }
C
chengduoZH 已提交
103 104

  // Col2Vol test
C
chengduoZH 已提交
105
  float col_2_vol[] = {0, 2, 2, 3, 8, 5, 6, 14, 8, 9, 20, 11};
C
chengduoZH 已提交
106 107 108 109 110 111 112 113
  memset(input_ptr, 0, 12 * sizeof(float));
  if (paddle::platform::is_cpu_place(*place)) {
    input = input_tmp;
  } else {
    input.CopyFrom<float>(input_tmp, *place);
  }

  paddle::operators::math::Col2VolFunctor<Place, float> col2vol;
C
chengduoZH 已提交
114
  col2vol(*context, input, output, stride, stride, stride, padding, padding,
C
chengduoZH 已提交
115 116
          padding);

C
chengduoZH 已提交
117
  float* in_ptr;
C
chengduoZH 已提交
118
  if (paddle::platform::is_cpu_place(*place)) {
C
chengduoZH 已提交
119
    in_ptr = input.data<float>();
C
chengduoZH 已提交
120 121
  } else {
    input_tmp.CopyFrom<float>(input, paddle::platform::CPUPlace());
C
chengduoZH 已提交
122
    in_ptr = input_tmp.data<float>();
C
chengduoZH 已提交
123 124
  }

C
chengduoZH 已提交
125
  for (int i = 0; i < 12; ++i) {
C
chengduoZH 已提交
126
    EXPECT_EQ(in_ptr[i], col_2_vol[i]);
C
chengduoZH 已提交
127
  }
C
chengduoZH 已提交
128 129 130 131
}

TEST(math, vol2col) {
  testVol2col<paddle::platform::CPUPlace>();
C
chengduoZH 已提交
132
#ifdef PADDLE_WITH_CUDA
C
chengduoZH 已提交
133
  testVol2col<paddle::platform::GPUPlace>();
C
chengduoZH 已提交
134
#endif  // PADDLE_WITH_CUDA
C
chengduoZH 已提交
135
}