Make 'scale_op' supporting int and int64 (#5986)

* Make 'scale_op' supporting int and int64 * refine .cu file

Make 'scale_op' supporting int and int64 (#5986)
* Make 'scale_op' supporting int and int64 * refine .cu file
23b3fef0 · fengjiayi · GitHub · 6375c8ca · 23b3fef0 · 23b3fef0
隐藏空白更改
内联并排

Showing with 6 addition and 2 deletion

paddle/operators/scale_op.cc paddle/operators/scale_op.cc +3 -1

paddle/operators/scale_op.cu paddle/operators/scale_op.cu +3 -1

未找到文件。
--- a/paddle/operators/scale_op.cc
+++ b/paddle/operators/scale_op.cc
@@ -77,4 +77,6 @@ REGISTER_OPERATOR(scale, ops::ScaleOp, ops::ScaleOpMaker<float>,
                  ops::ScaleGradMaker);
 REGISTER_OP_CPU_KERNEL(scale,
                       ops::ScaleKernel<paddle::platform::CPUPlace, float>,
-                       ops::ScaleKernel<paddle::platform::CPUPlace, double>);
+                       ops::ScaleKernel<paddle::platform::CPUPlace, double>,
+                       ops::ScaleKernel<paddle::platform::CPUPlace, int>,
+                       ops::ScaleKernel<paddle::platform::CPUPlace, int64_t>);
--- a/paddle/operators/scale_op.cu
+++ b/paddle/operators/scale_op.cu
@@ -16,4 +16,6 @@

 REGISTER_OP_GPU_KERNEL(
    scale, paddle::operators::ScaleKernel<paddle::platform::GPUPlace, float>,
-    paddle::operators::ScaleKernel<paddle::platform::GPUPlace, double>);
+    paddle::operators::ScaleKernel<paddle::platform::GPUPlace, double>,
+    paddle::operators::ScaleKernel<paddle::platform::GPUPlace, int>,
+    paddle::operators::ScaleKernel<paddle::platform::GPUPlace, int64_t>);