Deleting the sync operation and gpu.place check function in forward k…

…ernel.
PaddlePaddle · Feb 3, 2021 · aadcd13 · aadcd13 · paddle-bot-old · Feb 3, 2021
1 parent ce289f4
commit aadcd13
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/paddle/fluid/operators/index_sample_op.cu b/paddle/fluid/operators/index_sample_op.cu
@@ -81,10 +81,6 @@ class IndexSampleKernel<platform::CUDADeviceContext, T>
                               framework::proto::VarType::INT32),
                           paddle::framework::DataTypeToString(
                               framework::proto::VarType::INT64)));
-    PADDLE_ENFORCE_EQ(
-        platform::is_gpu_place(ctx.GetPlace()), true,
-        platform::errors::InvalidArgument("It must use CUDAPlace."));
-
     const auto* in_data = input->data<T>();
     auto* out_data = output->mutable_data<T>(ctx.GetPlace());
     auto stream =
@@ -115,7 +111,6 @@ class IndexSampleKernel<platform::CUDADeviceContext, T>
           index_data, in_data, out_data, index_length, input_length,
           batch_size);
     }
-    PADDLE_ENFORCE_CUDA_SUCCESS(cudaStreamSynchronize(stream));
   }
 };