From a89c2bcf530b0e4e3cb42d1de6816a1df357cad9 Mon Sep 17 00:00:00 2001 From: tensor-tang Date: Thu, 13 Jun 2019 14:54:58 +0800 Subject: [PATCH] fix and speedup ci conv test --- paddle/fluid/lite/core/op_registry.h | 3 --- .../fluid/lite/kernels/arm/conv_compute_test.cc | 15 +++++++++++++++ 2 files changed, 15 insertions(+), 3 deletions(-) diff --git a/paddle/fluid/lite/core/op_registry.h b/paddle/fluid/lite/core/op_registry.h index 8f5bd651689..1052419ecda 100644 --- a/paddle/fluid/lite/core/op_registry.h +++ b/paddle/fluid/lite/core/op_registry.h @@ -153,12 +153,9 @@ class KernelRegistor : public lite::Registor { public: KernelRegistor(const std::string &op_type, const std::string &alias) : Registor([=] { -<<<<<<< HEAD -======= VLOG(3) << "Register kernel " << op_type << " for " << TargetToStr(target) << " " << PrecisionToStr(precision) << " " << DataLayoutToStr(layout) << " alias " << alias; ->>>>>>> gitlab/develop KernelRegistry::Global().Register( op_type, [=]() -> std::unique_ptr { std::unique_ptr x(new KernelType); diff --git a/paddle/fluid/lite/kernels/arm/conv_compute_test.cc b/paddle/fluid/lite/kernels/arm/conv_compute_test.cc index 4b95aa5ce4a..7950788d300 100644 --- a/paddle/fluid/lite/kernels/arm/conv_compute_test.cc +++ b/paddle/fluid/lite/kernels/arm/conv_compute_test.cc @@ -124,6 +124,7 @@ TEST(conv_arm, init) { TEST(conv_arm, compute) { DeviceInfo::Init(); +#if 0 for (auto n : {1, 2}) { for (auto ic : {6, 32 /*, 128*/}) { for (auto oc : {6, 32 /*, 128*/}) { @@ -136,6 +137,20 @@ TEST(conv_arm, compute) { for (auto stride : {1, 2}) { for (auto padding : {0, 1, 2}) { for (auto ks : {1, 3, 5}) { +#else + for (auto n : {1}) { + for (auto ic : {6}) { + for (auto oc : {6}) { + for (auto ih : {9}) { + for (auto iw : {9}) { + for (auto flag_bias : {false, true}) { + for (auto flag_relu : {false, true}) { + for (auto depthwise : {false, true}) { + for (auto dilation : {1}) { + for (auto stride : {1}) { + for (auto padding : {0, 1}) { + for (auto ks : {1, 3, 5}) { +#endif int group = 1; if (depthwise) { // depthwise convolution ? group = oc = ic; -- GitLab