根据推力根据索引更改某些元素的值

Question

• 问题描述：我想使用thrust根据另一个index数组设置数组中某些元素的值。下面是一个简单的示例：

// input:
int array[10] = {1,9,9,6,0,1,1,6,1,4};
int set_num = -1;
int index[4] = {0,2,4,6};
// output:
array[10] = {-1,9,-1,6,-1,1,-1,6,1,4};

• 我的想法：我不知道如何使用thrust完成它，所以我编写了自己的内核。该代码段如下所示：

#include <cstdlib>
#include <iostream>
#include <cstdio>

#include "cuda_runtime.h"
#include "device_launch_parameters.h"


#include <thrust/host_vector.h>
#include <thrust/device_vector.h>
#include <thrust/execution_policy.h>
#include <thrust/copy.h>
#include <thrust/sequence.h>
#include <thrust/functional.h>

// kernel function
template <typename T>
__global__ void setNumInArray(T *arrays, int *index, T set_num, int num_index)
{
    unsigned int tid = threadIdx.x + blockDim.x * blockIdx.x;
    if (tid > num_index)
        return;
    arrays[index[tid]] = set_num;
}

int main()
{
    const unsigned int vec_length = 4069;                             // length of the array
    thrust::device_vector<int>vec(vec_length);
    thrust::sequence(thrust::device, vec.begin(), vec.end(), 1);      // the values are generated by thrust::sequence
    const int num_index = 16;                                         // length of the index array
    thrust::device_vector<int>index(num_index);                       // the index array is generated by thrust::sequence
    thrust::sequence(thrust::device, index.begin(), index.end(), 64,10);

    // type convert: thrust->normal device pointer
    int *d_vec = reinterpret_cast<int*>(thrust::raw_pointer_cast(vec.data()));
    int *d_index = reinterpret_cast<int*>(thrust::raw_pointer_cast(index.data()));

    int set_num = 0;    // the value I want to set

    int block_size = 64;
    setNumInArray<int> <<<1, block_size >>> (d_vec, d_index, set_num, num_index);

    thrust::host_vector<int>h_vec(vec_length);                         // check the result
    h_vec = vec;
    for (int ii = 0; ii < vec_length; ii++) {
        printf("%d: %d\n",ii+1,h_vec[ii]);
    }

    return 0;
}

该程序可以输出正确的答案。但是，我想避免在推力变量和通用设备变量之间进行类型转换（我在项目的另一部分中使用了很多推力函数），并且还避免了调整块/网格大小以获得更好的性能。那么如何使用thrust实现它？

• 工作环境：
  • 操作系统：Windows10
  • GPU：RTX2060
  • CUDA：10.2
  • IDE：VS2015
• 其他问题：
  1. 代码段中的类型转换方法正确吗？会引起一些潜在的问题吗？
  2. 为什么编译器（VS2015）找不到cudaOccupancyMaxPotentialBlockSizes？我想用它来确定最佳的块大小。

Answer 1

您可以将推力::: for_each与lambda函数一起使用。但是我还没有对此进行基准测试，也许您的内核更快。

int main()
{
    const unsigned int vec_length = 4069;                        
    thrust::device_vector<int>vec(vec_length);
    thrust::sequence(thrust::device, vec.begin(), vec.end(), 1);     
    const int num_index = 16;                                         
    thrust::device_vector<int>index(num_index);                       
    thrust::sequence(thrust::device, index.begin(), index.end(), 64,10);
    int *d_vec = thrust::raw_pointer_cast(vec.data());
    int set_num = 0;

    auto changeValue = [=] __device__(int y) { d_vec[y] = set_num; };
    thrust::for_each(thrust::device, index.begin(), index.end(), changeValue);

    thrust::host_vector<int>h_vec(vec_length);
    h_vec = vec;
    for (int ii = 0; ii < vec_length; ii++) {
        if(h_vec[ii] == set_num)
            printf("%d: %d\n",ii+1,h_vec[ii]);
    }
    return 0;
}