我有一个CUDA推力程序
#include <stdio.h>
#include<iostream>
#include <cuda.h>
#include <thrust/sort.h>
// main routine that executes on the host
int main(void)
{
int *a_h, *a_d; // Pointer to host & device arrays
const int N = 10; // Number of elements in arrays
size_t size = N * sizeof(int);
a_h = (int *)malloc(size); // Allocate array on host
cudaMalloc((void **) &a_d, size);// Allocate array on device
std::cout<<"enter the 10 numbers";
// Initialize host array and copy it to CUDA device
for (int i=0; i<N; i++)
{
std::cin>>a_h[i];
}
for (int i=0; i<N; i++) printf("%d %d\n", i, a_h[i]);
cudaMemcpy(a_d, a_h, size, cudaMemcpyHostToDevice);
thrust::sort(a_d, a_d + N);
// Do calculation on device:
cudaMemcpy(a_h, a_d, sizeof(int)*N, cudaMemcpyDeviceToHost);
// Print results
for (int i=0; i<N; i++) printf("%d %d\n", i, a_h[i]);
// Cleanup
free(a_h); cudaFree(a_d);
}
但无法运行以提供所需的输出。
我们是否应该使用宿主向量和设备向量对推力进行排序?
最佳答案
对于设备操作,您应该使用设备指针或device_vector迭代器,而不是原始指针。原始指针(指向主机内存)可用于主机上的操作。
因此,如果您按以下方式修改代码:
#include <thrust/device_ptr.h>
...
thrust::device_ptr<int> t_a(a_d); // add this line before the sort line
thrust::sort(t_a, t_a + N); // modify your sort line
我相信它将为您服务。
您可能希望阅读推力quick start guide。特别注意本节:
您可能想知道当将“原始”指针用作Thrust函数的参数时会发生什么。像STL一样,Thrust允许这种用法,它将分派算法的主机路径。如果所讨论的指针实际上是指向设备内存的指针,则在调用该函数之前,需要使用推力:: device_ptr对其进行包装
关于cuda - 使用推力进行简单分类不起作用,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/25242790/