diff --git a/src/operators/fusion_conv_add.h b/src/operators/fusion_conv_add.h index 4cfd85e476bbd6684d8fdb1b06068648506fe945..170df9ce33e4ab90297664fbc81d723e7c246f83 100644 --- a/src/operators/fusion_conv_add.h +++ b/src/operators/fusion_conv_add.h @@ -67,7 +67,7 @@ class FusionConvAddOp : public framework::OperatorWithKernel< #ifdef PADDLE_MOBILE_CPU #ifndef CONV_ADD_REGISTER - static framework::FusionOpRegistrar convadd_registrar( +static framework::FusionOpRegistrar convadd_registrar( new FusionConvAddMatcher()); #define CONV_ADD_REGISTER #endif diff --git a/test/net/test_mobilenet+ssd.cpp b/test/net/test_mobilenet+ssd.cpp index 7218d681d811ae63b76b8a7662bbe46f835ef00a..a3aac63f5759923df5bc60df556241c6e15c3eb4 100644 --- a/test/net/test_mobilenet+ssd.cpp +++ b/test/net/test_mobilenet+ssd.cpp @@ -19,8 +19,9 @@ limitations under the License. */ int main() { paddle_mobile::PaddleMobile paddle_mobile; auto time1 = time(); -// auto isok = paddle_mobile.Load(g_mobilenet_ssd_gesture + "/model", -// g_mobilenet_ssd_gesture + "/params", true); + // auto isok = paddle_mobile.Load(g_mobilenet_ssd_gesture + "/model", + // g_mobilenet_ssd_gesture + "/params", + // true); auto isok = paddle_mobile.Load(g_mobilenet_ssd, false); if (isok) { auto time2 = time(); @@ -36,7 +37,8 @@ int main() { auto time3 = time(); paddle_mobile.Predict(input, dims); auto time4 = time(); - std::cout << "predict cost :" << time_diff(time3, time4) << "ms" << std::endl; + std::cout << "predict cost :" << time_diff(time3, time4) << "ms" + << std::endl; } return 0; }