Fix error in SP

This commit is contained in:
Anthony Scemama 2023-08-21 13:23:18 +02:00
parent 095053333a
commit 0e768f7d4d
1 changed files with 2 additions and 4 deletions

View File

@ -37,8 +37,7 @@ void gpu_upload_sp(gpu_data_sp* data,
for (size_t i=0 ; i<(size_t) nO*nV ; ++i) {
AA[i] = t1[i];
}
cudaStat = cublasSetMatrix(nO, nV, sizeof(float), AA, lda, d_t1, lda);
assert (cudaStat == cudaSuccess);
cublasSetMatrix(nO, nV, sizeof(float), AA, lda, d_t1, lda);
free(AA);
float* d_t2 = data[igpu].t2;
@ -48,8 +47,7 @@ void gpu_upload_sp(gpu_data_sp* data,
for (size_t i=0 ; i<(size_t) nO*nO*nV*nV ; ++i) {
AA[i] = t2[i];
}
cudaStat = cublasSetMatrix(nO*nO, nV*nV, sizeof(float), AA, lda, d_t2, lda);
assert (cudaStat == cudaSuccess);
cublasSetMatrix(nO*nO, nV*nV, sizeof(float), AA, lda, d_t2, lda);
free(AA);
size_t lda, ldb, ldc;