From 5e4160d8ea0f5889185d386d9eb296352d1c9623 Mon Sep 17 00:00:00 2001 From: yangfei Date: Wed, 5 Dec 2018 11:43:19 +0800 Subject: [PATCH] imp dropout op kernel for gpu --- src/operators/kernel/cl/mul_kernel.cpp | 4 +--- src/operators/mul_op.cpp | 3 +++ 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/src/operators/kernel/cl/mul_kernel.cpp b/src/operators/kernel/cl/mul_kernel.cpp index 3edfd74c13..c59483c14f 100644 --- a/src/operators/kernel/cl/mul_kernel.cpp +++ b/src/operators/kernel/cl/mul_kernel.cpp @@ -25,9 +25,7 @@ bool MulKernel::Init(MulParam *param) { } template <> -void MulKernel::Compute(const MulParam ¶m) { - -} +void MulKernel::Compute(const MulParam ¶m) {} template class MulKernel; diff --git a/src/operators/mul_op.cpp b/src/operators/mul_op.cpp index 69e3bb300d..ec9c8e2254 100644 --- a/src/operators/mul_op.cpp +++ b/src/operators/mul_op.cpp @@ -58,6 +58,9 @@ namespace ops = paddle_mobile::operators; #ifdef PADDLE_MOBILE_CPU REGISTER_OPERATOR_CPU(mul, ops::MulOp); #endif +#ifdef PADDLE_MOBILE_CL +REGISTER_OPERATOR_CL(mul, ops::MulOp); +#endif #ifdef PADDLE_MOBILE_MALI_GPU REGISTER_OPERATOR_MALI_GPU(mul, ops::MulOp); #endif -- GitLab