jiri/LA_library
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

std::complex in cuda_la.cc and noncblas.cc

Browse Source
This commit is contained in:
Jiri Pittner 2023-04-28 15:49:18 +02:00
parent 984c37721a
commit 5f1dd5f6e8
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
cuda_la.cc
View File

@ -48,15 +48,15 @@ double *gpuputdouble(const double &x){
return (double *)ptr;
}
complex<double> *gpuputcomplex(const complex<double> &x){
std::complex<double> *gpuputcomplex(const std::complex<double> &x){
void *ptr = NULL;
cublasAlloc(1, sizeof(complex<double>), &ptr);
cublasAlloc(1, sizeof(std::complex<double>), &ptr);
TEST_CUBLAS("cublasAlloc");
cublasSetVector(1, sizeof(complex<double>), &x, 1, ptr, 1);
cublasSetVector(1, sizeof(std::complex<double>), &x, 1, ptr, 1);
TEST_CUBLAS("cublasSetVector");
return (complex<double> *)ptr;
return (std::complex<double> *)ptr;
}

6
noncblas.cc
View File

@ -255,8 +255,8 @@ void cblas_zgerc(const CBLAS_ORDER Order, const int M, const int N,
{
if(Order!=CblasRowMajor) laerror("CblasRowMajor order asserted");
//conjugate y
complex<double> *p;
p= (complex<double> *) Y; for(int i=0; i<N; ++i) {p->imag() = -p->imag(); p+= incY;}
std::complex<double> *p;
p= (std::complex<double> *) Y; for(int i=0; i<N; ++i) {p->imag(-p->imag()); p+= incY;}
#ifdef FORINT
const FINT mtmp=M;
const FINT ntmp=N;
@ -268,7 +268,7 @@ p= (complex<double> *) Y; for(int i=0; i<N; ++i) {p->imag() = -p->imag(); p+= in
FORNAME(zgerc)(&N, &M, alpha, Y, &incY, X, &incX, A, &lda);
#endif
//conjugate y back
p= (complex<double> *) Y; for(int i=0; i<N; ++i) {p->imag() = -p->imag(); p+= incY;}
p= (std::complex<double> *) Y; for(int i=0; i<N; ++i) {p->imag(--p->imag()); p+= incY;}
}