CUDA - Cách làm việc với các số phức?

Question

Tôi nên bao gồm tiêu đề CUDA nào trong chương trình nếu tôi muốn làm việc với số phức và thực hiện các phép toán đơn giản (cộng và phép nhân) với các số phức kép này trong chính hạt nhân?

Trong C++, tôi có thể nhân một số không đổi với số phức kép> miễn là chúng đều là hai lần. Tuy nhiên trong CUDA tôi nhận được rất nhiều lỗi khi tôi cố gắng thực hiện các phép toán đơn giản thành phức tạp gấp đôi> s bất cứ khi nào nó không phải với một double phức tạp khác>. Tôi đang thiếu gì?

Cảm ơn bạn!

Answer 1

Phần header bao gồm là:

#include <cuComplex.h> 

Trên một CUDA linux tiêu chuẩn cài đặt, nó nằm ở:

/usr/local/cuda/include 

Bạn sẽ cần phải kiểm tra rằng tập tin header và sử dụng các chức năng quy định tại nó để thao tác các số phức trên thiết bị.

Để nhân một (double) số phức tạp bởi một số thực, tôi sẽ:

#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: Với v1.8 Thrust gần đây phát hành vào CUDA 7 RC, nó có thể sử dụng lực đẩy :: phức tạp trong mã lực đẩy hoặc Mã thiết bị CUDA. Điều này giúp bạn có thể viết nhiều hoạt động trông tự nhiên hơn như:

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