未验证 提交 1cbb282d 编写于 作者: H Huihuang Zheng 提交者: GitHub

Add Retry Logic to CublasHandlerHolder

Add Retry Logic to CublasHandlerHolder to avoid random unittest failure.
上级 96934b74
......@@ -86,7 +86,7 @@ class CublasHandleHolder {
dynload::cublasSetMathMode(handle_, CUBLAS_TENSOR_OP_MATH));
#if CUDA_VERSION >= 11000
} else if (math_type == CUBLAS_TF32_TENSOR_OP_MATH) {
PADDLE_ENFORCE_CUDA_SUCCESS(
PADDLE_RETRY_CUDA_SUCCESS(
dynload::cublasSetMathMode(handle_, CUBLAS_TF32_TENSOR_OP_MATH));
#endif // CUDA_VERSION >= 11000
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册