Fix rocm error.

PaddlePaddle · Mar 2, 2022 · 7b22822 · 7b22822 · paddle-bot-old · Mar 2, 2022
1 parent 6b0d04e
commit 7b22822
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/fluid/operators/dropout_impl.cu.h b/paddle/fluid/operators/dropout_impl.cu.h
@@ -259,7 +259,7 @@ void DropoutFwGPUKernelDriver(const platform::CUDADeviceContext& dev_ctx,
 #ifdef PADDLE_WITH_HIP
       PADDLE_ENFORCE_GPU_SUCCESS(
           hipMemcpyAsync(y_data, x_data, sizeof(T) * x_numel,
-                         cudaMemcpyDeviceToDevice, stream));
+                         hipMemcpyDeviceToDevice, stream));
 #else
       PADDLE_ENFORCE_GPU_SUCCESS(
           cudaMemcpyAsync(y_data, x_data, sizeof(T) * x_numel,