程序员参考/文档

Question

我正在做一个袖带实现，找不到任何关于袖带复杂函数的引用。不过，我通过谷歌找到了cucomp.h，但这对我没什么帮助。具体来说，我想知道，如何读出袖口结构的虚部和实部。

Answer 1

cufftComplex和cuComplex的类型实际上是相同的。它被记录在cuFFT文档中。在cufft.h中，您可以找到typedef：

typedef cuComplex cufftComplex;

在cuComplex.h中，您会发现cuComplex确实是一个float2，也就是说，您可以用c.x读出真实值，用c.y读取虚值。或者更好地使用cuCrealf()和cuCimagf()提供的cuComplex.h函数。