将 Typename1 的 Cuda Array1 转换为 Typename2 的 Array2

Question

亲爱的 Cuda 学者，
寻找以下问题的解决方案

a）我有两个数组 1) size1 的 array1，其类型为 name1 2) array2 of size1 is of typename2

b) 我想编写以下原型的内核

__global__ 内核(void* dest, void* src, int dest_sizeoftype, int src_sizeoftype, int num_array_elts);

c) 假设我创建 num_array_elts 个 cuda 线程，每个线程将其 elt 从 src 复制到目标。

问题： a) 我遇到困难的地方是使用哪个函数将 num_bytes 从 src 复制到内核中的 dest 。

预先感谢您 问候， 纳加拉朱

Answer 1

Thrust 中的复制算法使这一切变得简单。

#include <thrust/copy.h>
#include <thrust/device_ptr.h>

int   * src = ...
float * dst = ...

// first wrap the 'raw' pointers
thrust::device_ptr<int>   wrapped_src(src);
thrust::device_ptr<float> wrapped_dst(dst);

// then pass wrapped pointers to copy()
thrust::copy(wrapped_src, wrapped_src + num_array_elts, wrapped_dst);

有关 Thrust 的更多信息，请参阅快速入门指南。

Answer 2

如果您知道两个数组的类型，这个问题就变得相当简单。

__global__ kernel(float* dest, int* src){

    int idx=blockIdx.x*blockDim.x+threadIdx.x;
    dest[ idx ] = src[ idx ];

}

如果您的目标数组使用更大的字，例如双精度数，这仍然有效，并且不需要知道字节数。只需确保在使用 cudamalloc。