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