You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
3.1 KiB
100 lines
3.1 KiB
/* 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 <array>
|
|
#include <vector>
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include "paddle/framework/data_transform.h"
|
|
|
|
namespace paddle {
|
|
namespace framework {
|
|
using namespace platform;
|
|
|
|
/**
|
|
* @brief cross validation of different kernel type transform
|
|
* We use four bit map represent different combination.
|
|
* If the field has multiple possible value, only choose two of them.
|
|
* For DataType, only test the FP32(float), FP64(double).
|
|
* e.g. 0000 -> FP32, CPUPlace, kNHWC, kPlain
|
|
* 1111 -> FP64, GPUPlace, kNCHW, kMKLDNN
|
|
*/
|
|
|
|
std::array<proto::DataType, 2> kDataType = {
|
|
{proto::DataType::FP32, proto::DataType::FP64}};
|
|
|
|
std::array<Place, 2> kPlace = {{CPUPlace(), CUDAPlace(0)}};
|
|
|
|
std::array<DataLayout, 2> kDataLayout = {
|
|
{DataLayout::kNHWC, DataLayout::kNCHW}};
|
|
|
|
std::array<LibraryType, 2> kLibraryType = {
|
|
{LibraryType::kPlain, LibraryType::kMKLDNN}};
|
|
|
|
OpKernelType GenFromBit(const std::vector<bool> bits) {
|
|
return OpKernelType(kDataType[bits[0]], kPlace[bits[1]], kDataLayout[bits[2]],
|
|
kLibraryType[bits[3]]);
|
|
}
|
|
|
|
int test_value = 0;
|
|
|
|
auto kernel0 = GenFromBit({0, 0, 0, 0});
|
|
auto kernel1 = GenFromBit({0, 0, 0, 1});
|
|
auto kernel2 = GenFromBit({0, 0, 1, 0});
|
|
auto kernel3 = GenFromBit({0, 0, 1, 1});
|
|
|
|
void TransDataType_t(const platform::DeviceContext* ctx, const Variable& in,
|
|
Variable* out) {
|
|
test_value++;
|
|
}
|
|
|
|
void TransDataLayout_t(const platform::DeviceContext* ctx, const Variable& in,
|
|
Variable* out) {
|
|
test_value--;
|
|
}
|
|
|
|
void TransLibraryType_t(const platform::DeviceContext* ctx, const Variable& in,
|
|
Variable* out) {
|
|
test_value += 2;
|
|
}
|
|
|
|
} // namespace framework
|
|
} // namespace paddle
|
|
|
|
namespace frw = paddle::framework;
|
|
|
|
REGISTER_DATA_TRANSFORM_FN(frw::kernel0, frw::kernel1, frw::TransDataType_t);
|
|
REGISTER_DATA_TRANSFORM_FN(frw::kernel1, frw::kernel2, frw::TransDataLayout_t);
|
|
REGISTER_DATA_TRANSFORM_FN(frw::kernel0, frw::kernel2, frw::TransLibraryType_t);
|
|
|
|
TEST(DataTransform, Register) {
|
|
using namespace paddle::framework;
|
|
using namespace paddle::platform;
|
|
|
|
auto& instance = DataTransformFnMap::Instance();
|
|
ASSERT_EQ(instance.Map().size(), 3UL);
|
|
DeviceContext* ctx = nullptr;
|
|
paddle::framework::Variable in;
|
|
paddle::framework::Variable out;
|
|
|
|
instance.Get(std::make_pair(frw::kernel0, frw::kernel1))(ctx, in, &out);
|
|
ASSERT_EQ(test_value, 1);
|
|
|
|
instance.Get(std::make_pair(frw::kernel1, frw::kernel2))(ctx, in, &out);
|
|
ASSERT_EQ(test_value, 0);
|
|
|
|
instance.Get(std::make_pair(frw::kernel0, frw::kernel2))(ctx, in, &out);
|
|
ASSERT_EQ(test_value, 2);
|
|
}
|