CUDA - 如何處理複雜的數字？

Question

如果我想處理複數並在內核中對這些複數雙數進行簡單的數學運算（加法和乘法），應該在程序中包含哪些CUDA頭文件？

在C++中，我可以乘以一個複雜的雙恆定數量>只要它們都加倍。然而，在CUDA中，當我嘗試執行簡單的數學運算來複雜double> s時，我會遇到很多錯誤，只要它不與另一個複合double>。我錯過了什麼？

謝謝！

Answer 1

頭包括爲：

#include <cuComplex.h> 

在一個標準的Linux CUDA安裝，它位於：

/usr/local/cuda/include 

您需要檢查該頭文件，並使用規定的職責它操縱設備上的複數。

要按實數乘以（雙）複數，我想：

#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)); 

編輯：隨着CUDA 7 RC剛剛發佈的推力V1.8，可以使用推力::複合推力碼或 CUDA設備代碼。這樣可以編寫更自然的操作，例如：

#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();