mirror of
https://gitlab.com/scemama/qp_plugins_scemama.git
synced 2024-08-29 23:43:43 +02:00
Fix error in SP
This commit is contained in:
parent
095053333a
commit
0e768f7d4d
@ -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;
|
||||||
|
Loading…
Reference in New Issue
Block a user