mirror of
https://gitlab.com/scemama/qp_plugins_scemama.git
synced 2024-08-29 15:33:40 +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) {
|
||||
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;
|
||||
|
Loading…
Reference in New Issue
Block a user