add_op.cu 251 字节
Newer Older
Q
qijun 已提交
1 2
#include "paddle/operators/add_op.h"
#include "paddle/framework/op_registry.h"
Y
Yu Yang 已提交
3

Q
qijun 已提交
4
typedef paddle::operators::AddKernel<::paddle::platform::GPUPlace, float> AddKernel_GPU_float;
Y
Yu Yang 已提交
5
REGISTER_OP_GPU_KERNEL(add_two,
Q
qijun 已提交
6
                       AddKernel_GPU_float);