scemama/qp_plugins_scemama
1
0
Fork 0
mirror of https://gitlab.com/scemama/qp_plugins_scemama.git synced 2025-01-03 01:55:52 +01:00
Code Issues Releases Wiki Activity

GPU acceleration

Browse Source
This commit is contained in:
Anthony Scemama 2023-07-16 13:50:40 +02:00
parent 7847ccc674
commit 0c6e6a1ca0
1 changed files with 15 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
devel/ccsd_gpu/gpu.c
View File

@ -105,7 +105,8 @@ void compute_r2_space_chol_gpu(const int nO, const int nV, const int cholesky_mo
C=d_tmpB1 ; ldc=nV*BLOCK_SIZE;
cublasDgemm(handle, CUBLAS_OP_T, CUBLAS_OP_N, m, n, k, &alpha, A, lda, B, lda, &beta, C, ldc);
for (size_t bet=iblock ; bet<(nV < iblock+BLOCK_SIZE ? nV : iblock+BLOCK_SIZE) ; ++bet) {
for (size_t bet=iblock ; bet<(nV < iblock+BLOCK_SIZE ? nV : iblock+BLOCK_SIZE) ; ++bet)
{
alpha = 1.0;
beta = 0.0;
@ -113,6 +114,7 @@ void compute_r2_space_chol_gpu(const int nO, const int nV, const int cholesky_mo
B = d_tmpB1; ldb = nV;
C = &(d_B1[nV*nV*(bet-iblock)]) ; ldc = nV;
cublasDgeam(handle, CUBLAS_OP_N, CUBLAS_OP_N, nV, nV, &alpha, A, lda, &beta, B, ldb, C, ldc);
}
alpha=1.0; beta=1.0;
m=nO*nO; n=mbs; k=nV*nV;
@ -124,9 +126,6 @@ void compute_r2_space_chol_gpu(const int nO, const int nV, const int cholesky_mo
}
}
free(B1);
free(tmpB1);
}
lda=nO*nO;
cublasGetMatrix(nO*nO, nV*nV, sizeof(double), d_r2, lda, r2, lda);