linux – CUDA C v.推特,我错过了什么吗？

我觉得很难相信

__global__ void square(float *a,int N) {
    int idx = blockIdx.x * blockDim.x + threadIdx.x;
    if (idx < N) {
        a[idx] = a[idx] * a[idx];
    }
}

int main(int argc,char** argv) {

float *aHost,*aDevice;

const int N = 10;
size_t size = N * sizeof(float);

aHost = (float*)malloc(size);
cudaMalloc((void**)&aDevice,size);

for (int i = 0; i < N; i++) {
    aHost[i] = (float)i;
}

cudaMemcpy(aDevice,aHost,size,cudaMemcpyHostToDevice);

int block = 4;
int nBlock = N/block + (N % block == 0 ? 0:1);

square<<<nBlock,block>>>(aDevice,N);

cudaMemcpy(aHost,aDevice,cudaMemcpyDeviceToHost);

for (int i = 0; i < N; i++) {
    printf("%d,%fn",i,aHost[i]);
}

free(aHost);
cudaFree(aDevice);
}

是等价的

template <typename T>
    struct square {
    __host__ __device__ T operator()(const T& x) const {
        return x * x;
    }
}; 

int main(int argc,char** argv) {
    const int N = 10;
    thrust::device_vector<float> dVec(N);
    thrust::sequence(dVec.begin(),dVec.end());
    thrust::transform(dVec.begin(),dVec.end(),dVec.begin(),square<float>());
    thrust::copy(dVec.begin(),std::ostream_iterator<float>(std::cout,"n"));
}

我错过了什么吗？上面的代码是在GPU上运行的吗？ Thrust是一个很好的工具,但我怀疑它会处理所有繁重的C风格的内存管理.

> Thrust代码是否在GPU上执行？我该怎么说？
> Thrust是如何消除唤起内核的奇怪语法的？
> Thrust真的引起了内核吗？
> Thrust会自动处理线程索引计算吗？

谢谢你的时间.对不起,如果这些都是愚蠢的问题,但我发现我看到的例子很快就从可以被描述为Model T转变为M3.