ERNIE-Pytorch CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasGemmEx( handle, opa, opb, m, n, k, &falpha, a, CUDA_R_16F, lda, b, CUDA_R_16F, ldb, &fbeta, c, CUDA_R_16F, ldc, CUDA_R_32F, CUBLAS_GEMM_DFALT_TENSOR

CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasGemmEx( handle, opa, opb, m, n, k, &falpha, a, CUDA_R_16F, lda, b, CUDA_R_16F, ldb, &fbeta, c, CUDA_R_16F, ldc, CUDA_R_32F, CUBLAS_GEMM_DFALT_TENSOR_OP)

Open Julylmm opened this issue 2 years ago • 3 comments

When i loaded ernie-1.0 model to do fine-tune task, i came across runtime error. what's wrong?

Apr 14 '22 01:04 Julylmm

show me your load code

Apr 14 '22 02:04 nghuyong

我也遇到了一样的错误

Apr 25 '22 14:04 zh7y

@zh7y 能提供一个可以复现的sample code吗

Apr 25 '22 16:04 nghuyong