函数foo作为cuda核函数foo2的参数 foo的功能是将int src的赋值给int dst 请给出示例代码

下面是一个示例代码：

__global__ void foo2(int* dst, int* src, int size) {
    int tid = blockIdx.x * blockDim.x + threadIdx.x;
    if (tid < size) {
        foo(dst[tid], src[tid]);
    }
}

__device__ void foo(int& dst, int src) {
    dst = src;
}

int main() {
    int size = 100;
    int* src = new int[size];
    int* dst = new int[size];

    // 初始化src数组
    for (int i = 0; i < size; ++i) {
        src[i] = i;
    }

    // 将src数组拷贝到GPU的内存中
    int* d_src;
    cudaMalloc(&d_src, size * sizeof(int));
    cudaMemcpy(d_src, src, size * sizeof(int), cudaMemcpyHostToDevice);

    // 分配GPU内存，并调用核函数
    int* d_dst;
    cudaMalloc(&d_dst, size * sizeof(int));
    int threadsPerBlock = 256;
    int blocksPerGrid = (size + threadsPerBlock - 1) / threadsPerBlock;
    foo2<<<blocksPerGrid, threadsPerBlock>>>(d_dst, d_src, size);

    // 将结果拷贝回主机内存
    cudaMemcpy(dst, d_dst, size * sizeof(int), cudaMemcpyDeviceToHost);

    // 验证结果
    for (int i = 0; i < size; ++i) {
        assert(src[i] == dst[i]);
    }

    // 释放内存
    delete[] src;
    delete[] dst;
    cudaFree(d_src);
    cudaFree(d_dst);

    return 0;
}
``