diff --git a/paddle/fluid/platform/cuda_helper.h b/paddle/fluid/platform/cuda_helper.h index d6da830c9c4c7b53cba0d9054ef1a93a73b8f9f5..2a1f0b9ac5c4228ffa2ca790f3f3d06326803552 100644 --- a/paddle/fluid/platform/cuda_helper.h +++ b/paddle/fluid/platform/cuda_helper.h @@ -78,11 +78,11 @@ namespace platform { class CublasHandleHolder { public: CublasHandleHolder(cudaStream_t stream, cublasMath_t math_type) { - PADDLE_ENFORCE_CUDA_SUCCESS(dynload::cublasCreate(&handle_)); - PADDLE_ENFORCE_CUDA_SUCCESS(dynload::cublasSetStream(handle_, stream)); + PADDLE_RETRY_CUDA_SUCCESS(dynload::cublasCreate(&handle_)); + PADDLE_RETRY_CUDA_SUCCESS(dynload::cublasSetStream(handle_, stream)); #if CUDA_VERSION >= 9000 if (math_type == CUBLAS_TENSOR_OP_MATH) { - PADDLE_ENFORCE_CUDA_SUCCESS( + PADDLE_RETRY_CUDA_SUCCESS( dynload::cublasSetMathMode(handle_, CUBLAS_TENSOR_OP_MATH)); #if CUDA_VERSION >= 11000 } else if (math_type == CUBLAS_TF32_TENSOR_OP_MATH) { @@ -94,7 +94,7 @@ class CublasHandleHolder { } ~CublasHandleHolder() PADDLE_MAY_THROW { - PADDLE_ENFORCE_CUDA_SUCCESS(dynload::cublasDestroy(handle_)); + PADDLE_RETRY_CUDA_SUCCESS(dynload::cublasDestroy(handle_)); } template