gpt4 book ai didi

c++ - 来自复杂主机数据的 Arrayfire 数组

转载 作者:行者123 更新时间:2023-12-01 14:43:48 25 4
gpt4 key购买 nike

我在从主机数据初始化 arrayfire 数组时遇到问题。

以下代码不会为我链接:

constexp int mNumEigenInfos = 100;
std::complex<float> mEigenVectors[mNumEigenInfos][6];
af::array mEigenVectorsArray = af::array((dim_t)6,(dim_t)(mNumEigenInfos),reinterpret_cast<float2*>(mEigenVectors));

给我一个错误:
undefined reference to `af::array::array<float2>(long long, long long, float2 const*, af_source)'

现在,如果我将 reinterpret_cast 从 float2* 更改为 float*:
constexp int mNumEigenInfos = 100;
std::complex<float> mEigenVectors[mNumEigenInfos][6];
af::array mEigenVectorsArray = af::array((dim_t)6,(dim_t)(mNumEigenInfos),reinterpret_cast<float*>(mEigenVectors));

它链接很好。通过在线阅读,我认为我应该将复杂数据视为 cuComplex (将其转换为 float2 或 cuComplex 会给出完全相同的错误,因为它们是相同的)。

我觉得我在这里犯了一个愚蠢的错误,但似乎无法弄清楚。

我应该如何从 std::complex 主机数据初始化 arrayfire 数组?

感谢您的帮助

最佳答案

这有点类似于问题 af::array::device doesn't work with complex arrays请看一看 - 唯一的区别是传递复杂数据的方向。

以下应该可以正常工作 std::complex应该与 af::cfloat ABI 兼容.

af::array mEigenVectorsArray =
af::array((dim_t)6, (dim_t)(mNumEigenInfos),
reinterpret_cast<af::cfloat>(mEigenVectors));

ArrayFire API 将没有函数符号 float2类型,但是它将具有类型为 af::cfloat 的函数符号.偶 cuFloatComplextfloat2还应该与 af::cfloat 兼容 ABI我认为。因此,重新解释转换为 af::cfloat是应该做什么。

关于c++ - 来自复杂主机数据的 Arrayfire 数组,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59424886/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com