使用CUDA Thrust查找最大元素值及其位置

#include <thrust/device_vector.h>
#include <thrust/extrema.h>
#include <thrust/iterator/counting_iterator.h>

int main()
{
    // 初始化一个包含10个元素的设备向量
    thrust::device_vector<int> d_vec(10);

    // 填充向量
    d_vec[0] = 1;
    d_vec[1] = 3;
    d_vec[2] = 5;
    d_vec[3] = 2;
    d_vec[4] = 4;
    d_vec[5] = 6;
    d_vec[6] = 8;
    d_vec[7] = 7;
    d_vec[8] = 9;
    d_vec[9] = 0;

    // 查找最大元素值
    int max_val = *thrust::max_element(d_vec.begin(), d_vec.end());

    // 查找最大元素值的位置
    int max_idx = thrust::distance(d_vec.begin(), thrust::max_element(d_vec.begin(), d_vec.end()));

    // 输出结果
    std::cout << "Max value: "<< max_val<< std::endl;
    std::cout << "Max value index: "<< max_idx<< std::endl;

    return 0;
}