rowwise_add_op.cu 207 字节
Newer Older
1 2
#include "paddle/framework/op_registry.h"
#include "paddle/operators/rowwise_add_op.h"
3 4

REGISTER_OP_GPU_KERNEL(
Y
Yu Yang 已提交
5
    rowwise_add,
Q
qijun 已提交
6
    paddle::operators::RowWiseAddKernel<paddle::platform ::GPUPlace, float>);