CUDA - Come lavorare con numeri complessi?

Question

Quali intestazioni CUDA dovrei includere nel mio programma se voglio lavorare con numeri complessi e fare semplici operazioni matematiche (addizione e moltiplicazione) a questi numeri doppi complessi all'interno del kernel stesso?

In C++ posso moltiplicare un numero costante con un doppio complesso> purché siano entrambi doppi. Tuttavia in CUDA ho molti errori quando provo a fare semplici operazioni matematiche a complessi double> s ogni volta che non è con un altro doppio complesso>. Cosa mi manca?

Grazie!

Answer 1

L'intestazione da includere è:

#include <cuComplex.h> 

Su una CUDA Linux installazione standard, si trova in:

/usr/local/cuda/include 

Sarà necessario esaminare il file di intestazione e utilizzare le funzioni definite in per manipolare numeri complessi sul dispositivo.

per moltiplicare una (doppia) numero complesso da un numero reale, avrei:

#include <cuComplex.h> 
... 
double cr = 1; 
double ci = 2; 
double r = 3; 
cuDoubleComplex c = make_cuDoubleComplex(cr, ci); 
cuDoubleComplex result = cuCmul(c, make_cuDoubleComplex(r, 0)); 

EDIT: Con la spinta v1.8 recentemente rilasciato in CUDA 7 RC, è possibile utilizzare spinta :: complesso con codice di spinta o Codice dispositivo CUDA. Ciò rende possibile scrivere più operazioni dall'aspetto naturale come:

#include <thrust/complex.h> 
... 
thrust::complex<float> c = thrust::complex<float>(2.0f, 5.0f); 
thrust::complex<float> c2 = c*c; 
float r = c2.real();