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

5
REGISTER_OP_GPU_KERNEL(add_two, ops::AddKernel<ops::GPUPlace, float>);