Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 b851515b 编写于 作者: Z zchen0211
浏览文件
操作

merge new op grammar

上级 88a8eedd
隐藏空白更改
内联 并排
Showing with 4 addition and 9 deletion
paddle/operators/gather_op.cu
浏览文件 @ b851515b
......@@ -20,13 +20,8 @@
namespace paddle {
namespace operators {
// template <typename T>
__global__ void print_arr(const float *params, const int N) {
CUDA_1D_KERNEL_LOOP(i, N) { printf("device: %d, %f\n", i, params[i]); }
}
template <typename T>
class GatherOpCUDAKernel : public framework::OpKernel {
class GatherOpCUDAKernel : public framework::OpKernel<T> {
public:
void Compute(const framework::ExecutionContext &ctx) const override {
PADDLE_ENFORCE(platform::is_gpu_place(ctx.GetPlace()),
......@@ -42,7 +37,7 @@ class GatherOpCUDAKernel : public framework::OpKernel {
};
template <typename T>
class GatherGradOpCUDAKernel : public framework::OpKernel {
class GatherGradOpCUDAKernel : public framework::OpKernel<T> {
public:
void Compute(const framework::ExecutionContext &ctx) const override {
PADDLE_ENFORCE(platform::is_gpu_place(ctx.GetPlace()),
......
paddle/operators/scatter_op.cu
浏览文件 @ b851515b
......@@ -20,7 +20,7 @@ namespace paddle {
namespace operators {
template <typename T>
class ScatterOpCUDAKernel : public framework::OpKernel {
class ScatterOpCUDAKernel : public framework::OpKernel<T> {
public:
void Compute(const framework::ExecutionContext &ctx) const override {
PADDLE_ENFORCE(platform::is_gpu_place(ctx.GetPlace()),
......@@ -37,7 +37,7 @@ class ScatterOpCUDAKernel : public framework::OpKernel {
};
template <typename T>
class ScatterGradOpCUDAKernel : public framework::OpKernel {
class ScatterGradOpCUDAKernel : public framework::OpKernel<T> {
public:
void Compute(const framework::ExecutionContext &ctx) const override {
PADDLE_ENFORCE(platform::is_gpu_place(ctx.GetPlace()),
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或