add_op.cu 193 字节
Newer Older
Q
qiaolongfei 已提交
1
#include <paddle/operators/add_op.h>
Y
Yu Yang 已提交
2 3 4
#include <paddle/framework/op_registry.h>

REGISTER_OP_GPU_KERNEL(add_two,
Q
qiaolongfei 已提交
5
                       paddle::operators::AddKernel<paddle::platform::GPUPlace>);