RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)

[AAUfoa]

I cant run the test.py and train.py, errors are as follows:
RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)
How to fix this?

[gitouni]

if you don't type --use_rii when training and testing, this problem will be fixed. The cuda module file rri.cu may cause error if your cuda version is not the same as the author's. Hope the author will fix it soon.