使用推力的矢量数组

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>'