如何在cuda内核中添加向量元素?

时间:2012-10-21 21:59:59

标签: c++ cuda

#include <cuda.h>
#include <cuda_runtime.h>
#include <iostream>
#include <vector>
using namespace std;

__global__ void kernel(vector <int> *a,vector <int> *b, vector <int> *c)
{
    int index=threadIdx.x;
    c[index]=a[index]+b[index];
};

int main ()
{
    //Paar constanten initializeren
    int N=3;
    vector <int> a (3,1);
    vector <int> b (3,1);
    vector <int> c (3,0);
    int size=N*sizeof(vector <int>);
    //Device variabelen
    vector <int> d_a;
    vector <int> d_b;
    vector <int> d_c;

    //<size> bytes device-geheugen vrijmaken op locatie &a
    cudaMalloc( (void **) &a, size);
    cudaMalloc ( (void **) &b, size);
    cudaMalloc ( (void **) &c, size);

    //Kopieer inputs van &a naar &d_a
    cudaMemcpy(&d_a,&a,size,cudaMemcpyHostToDevice);
    cudaMemcpy(&d_b,&b,size,cudaMemcpyHostToDevice);

    printf( (char *) a[1]);
    printf( (char *) c[1]);
    //Voer kernel uit

    kernel<<<N,1>>>(&d_a,&d_b,&d_c);

    //Kopieer resultaat terug naar host 
    cudaMemcpy(&c,&d_c,size,cudaMemcpyDeviceToHost);

    printf( (char *) c[1]);

    return 0;
};

所以这是我的基本C ++向量添加程序,虽然它不会编译,因为它在添加索引时不知道如何处理“+”(“no operator”+“匹配这些操作数”)。任何人都知道为什么这在内核中不起作用?我认为它可能与它是一个主机功能或类似的事实有关,尽管我不知道如何解决这个问题。

提前问候和感谢。

1 个答案:

答案 0 :(得分:3)

你不能在设备代码中使用STL ......你将不得不使用数组或其他一些CUDA构造,如Thrust Library