CUDA - 복소수 사용법?

Question

복소수로 작업하고 커널 내에서 복잡한 복소수에 대해 간단한 수학 연산 (더하기 및 곱하기)을 수행하려면 어떤 CUDA 헤더를 프로그램에 포함시켜야합니까?

C++에서 상수에 double을 곱하면 복잡한 상수에 상수를 곱할 수 있습니다. 그러나 CUDA에서 다른 복잡한 double>이 없을 때마다 복잡한 double 연산에 간단한 연산을 시도 할 때 많은 오류가 발생합니다. 내가 뭘 놓치고 있니?

감사합니다. 포함하는

Answer 1

헤더는 다음과 같습니다

표준 리눅스 CUDA에

#include <cuComplex.h> 

설치, 그것은에 있습니다

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