From 86afb85907d74d8d5e6fbe5ca814decc03f4ab43 Mon Sep 17 00:00:00 2001 From: zchen0211 Date: Fri, 15 Sep 2017 22:49:43 -0700 Subject: [PATCH] prelu with gpu --- paddle/operators/prelu_op.cu | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 paddle/operators/prelu_op.cu diff --git a/paddle/operators/prelu_op.cu b/paddle/operators/prelu_op.cu new file mode 100644 index 000000000..9e391daba --- /dev/null +++ b/paddle/operators/prelu_op.cu @@ -0,0 +1,21 @@ +/* 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 "paddle/operators/prelu_op.h" + +REGISTER_OP_GPU_KERNEL( + prelu, paddle::operators::PReluKernel); +REGISTER_OP_GPU_KERNEL( + prelu_grad, + paddle::operators::PReluGradKernel); -- GitLab