Question

我的数据类型定义为typedef InitialState float[12]。我有一个包含由std::vector<InitialState> h_initials定义的几个初始状态的向量。

我把它作为推力设备矢量：thrust::device_vector<InitialState> d_initials = h_initials;

我想使用这个数据是用户定义的内核。但是，我在将其转换为原始指针时遇到问题。我试过了

float *p_initials = thrust::raw_pointer_cast(&d_initials[0]);

但thrust抱怨function returning array is not allowed。

有没有办法可以将设备向量转换为内核可以使用的指针？

Answer 1

InitialState（float [12]）！= float

InitialState *p_initials = thrust::raw_pointer_cast(d_initials.data());
float* p_floats = (float*)p_initials;

然而，由于下面的奇怪行为，这通常是错误的。

typedef int int4[4];
void test(int4 a)
{
   std::cout << sizeof(a) << std::endl;
}

int main(int argc, char** argv)
{
  int4 f;
  std::cout << sizeof(f) << std::endl;//returns 16 (4*sizeof(int))
  test(f);//returns 8 (size of pointer, passes by reference)
}

这更好的是：

struct InitialState{float value[12];}
std::vector<InitialState> h_initials;
thrust::device_vector<InitialState> d_initials = h_initials;
InitialState *p_initials = thrust::raw_pointer_cast(d_initials.data());
float* p_floats = (float*)p_initials;

在cuda中，您可以使用InitialState *或float *作为参数（尽管SOA比AOS更好）

2d thrust :: device_vector到内核

1 个答案: