randperm run error in multi-gpus (#27942)

zhupengyang · web-flow · commit 6dd64b0a3008 · 2020-10-20T16:14:02.000+08:00
diff --git a/paddle/fluid/operators/randperm_op.h b/paddle/fluid/operators/randperm_op.h
@@ -57,7 +57,7 @@ class RandpermKernel : public framework::OpKernel<T> {
       tmp_tensor.Resize(framework::make_ddim({n}));
       T* tmp_data = tmp_tensor.mutable_data<T>(platform::CPUPlace());
       random_permate<T>(tmp_data, n, seed);
-      framework::TensorCopy(tmp_tensor, platform::CUDAPlace(), out_tensor);
+      framework::TensorCopy(tmp_tensor, ctx.GetPlace(), out_tensor);
     }
   }
 };

Original file line number	Diff line number	Diff line change
`@@ -57,7 +57,7 @@ class RandpermKernel : public framework::OpKernel<T> {`
`57`	`57`	`tmp_tensor.Resize(framework::make_ddim({n}));`
`58`	`58`	`T* tmp_data = tmp_tensor.mutable_data<T>(platform::CPUPlace());`
`59`	`59`	`random_permate<T>(tmp_data, n, seed);`
`60`		`- framework::TensorCopy(tmp_tensor, platform::CUDAPlace(), out_tensor);`
	`60`	`+ framework::TensorCopy(tmp_tensor, ctx.GetPlace(), out_tensor);`
`61`	`61`	`}`
`62`	`62`	`}`
`63`	`63`	`};`