cuda - 使用推力的向量数组

Question

是否可以使用 Thrust 创建一个 device_vectors 数组？我知道我无法创建 device_vector 的 device_vector，但我将如何创建 device_vectors 数组？

Answer 1

以下代码对我有用。如果将此代码放在扩展名为 .cu 的文件中，则编译良好，但如果将其放在扩展名为 .cpp 的文件中，则会导致编译时断言失败。

thrust::device_vector<float> vectors[3];
//thrust::device_vector<float> *vectors = new thrust::device_vector<float>[3];

vectors[0] = thrust::device_vector<float>(10);
vectors[1] = thrust::device_vector<float>(10);
vectors[2] = thrust::device_vector<float>(10);

printf("Works\n");

断言失败如下

1>../for_each.inl(96) : error C2027: use of undefined type 'thrust::detail::STATIC_ASSERTION_FAILURE<x>'