scemama/qp_plugins_scemama
1
0
Fork 0
mirror of https://gitlab.com/scemama/qp_plugins_scemama.git synced 2024-12-22 12:23:37 +01:00
Code Issues Releases Wiki Activity

Fix error in SP

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
devel/ccsd_gpu/gpu_sp.c
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) { for (size_t i=0 ; i<(size_t) nO*nV ; ++i) {
AA[i] = t1[i]; AA[i] = t1[i];
} }
cudaStat = cublasSetMatrix(nO, nV, sizeof(float), AA, lda, d_t1, lda); cublasSetMatrix(nO, nV, sizeof(float), AA, lda, d_t1, lda);
assert (cudaStat == cudaSuccess);
free(AA); free(AA);
float* d_t2 = data[igpu].t2; 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) { for (size_t i=0 ; i<(size_t) nO*nO*nV*nV ; ++i) {
AA[i] = t2[i]; AA[i] = t2[i];
} }
cudaStat = cublasSetMatrix(nO*nO, nV*nV, sizeof(float), AA, lda, d_t2, lda); cublasSetMatrix(nO*nO, nV*nV, sizeof(float), AA, lda, d_t2, lda);
assert (cudaStat == cudaSuccess);
free(AA); free(AA);
size_t lda, ldb, ldc; size_t lda, ldb, ldc;