randperm run error in multi-gpus (#27942)

4 years ago · 6dd64b0a30
parent 74fadeb44a
commit 6dd64b0a30
1 changed files with 1 additions and 1 deletions
--- 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);
    }
  }
 };