提交 5e4160d8 编写于 作者: Y yangfei

imp dropout op kernel for gpu

上级 7dbb4fb8
......@@ -25,9 +25,7 @@ bool MulKernel<GPU_CL, float>::Init(MulParam<GPU_CL> *param) {
}
template <>
void MulKernel<GPU_CL, float>::Compute(const MulParam<GPU_CL> &param) {
}
void MulKernel<GPU_CL, float>::Compute(const MulParam<GPU_CL> &param) {}
template class MulKernel<GPU_CL, float>;
......
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册