CUDA - 推力::在设备上排序只返回 0

Question

我已经运行以下 Thrust 示例进行排序。 问题是，在thrust::sort ，输出包含所有0 。

请告诉我这里有什么问题。

#include <thrust/host_vector.h>
#include <thrust/device_vector.h>
#include <thrust/copy.h>    
#include <thrust/sort.h>
#include <cstdlib>
#include <iostream>

using namespace std;

int main(void)
{

    thrust::host_vector<int> h_vec(32 << 20);
    thrust::generate(h_vec.begin(), h_vec.end(), rand);


    thrust::device_vector<int> d_vec=h_vec;

    for(int i = 0; i<32;i++)
        cout<<d_vec[i]<<endl;

    cout<<endl<<endl<<endl;
    thrust::sort(d_vec.begin(), d_vec.end());

    for(int i = 0; i<32;i++)
        cout<<d_vec[i]<<endl;

    cout<<endl<<endl<<endl; 

    thrust::copy(d_vec.begin(), d_vec.end(), h_vec.begin());


    for(int i = 0; i<32;i++)
        cout<<h_vec[i]<<endl;


    return 0;
}

Answer 1

您观察所有0的原因是您正在生成大量随机数，即32 << 20 = 33554432 ，介于0和RAND_MAX之间，您正在对它们进行排序，但您只显示其中的32 。

正如 Robert Crovella 所提到的，在 Windows 机器上（OP 正在 Windows 上运行）， RAND_MAX = 2^15-1 = 32767 。 因此，您将生成0和32767之间的33554432整数，这意味着原始数组中将有大量0 ，因此排序数组的前32数字中全部为0 。

我已经亲自验证过，这在 Windows 32和64位机器上都会发生，也就是说，在 Windows 32和64位系统上RAND_MAX = 32767 。

同样，正如 Robert 所指出的，这种效果将在 Linux 32位机器上显示，但不会在 Linux 64位机器上显示，对于RAND_MAX = 2^31-1因为在这种情况下， RAND_MAX远大于32 << 20 .

正如罗伯特所建议的，可以改变指令

thrust::host_vector<int> h_vec(32 << 20);

到

thrust::host_vector<int> h_vec(min(32 << 20,RAND_MAX));

避免全0的显示。