将char数组传递给CUDA内核

时间:2014-09-24 07:49:13

标签: c++ c cuda arrays

我正在尝试将包含10000字的char数组从主函数中的txt文件读取到CUDA内核函数。

这些单词从主机传输到设备:

(主要功能代码:)

//.....
     const int text_length = 20;

     char (*wordList)[text_length] = new char[10000][text_length];
     char *dev_wordList;

     for(int i=0; i<number_of_words; i++)
     {
         file>>wordList[i];
         cout<<wordList[i]<<endl;
     }

     cudaMalloc((void**)&dev_wordList, 20*number_of_words*sizeof(char));
     cudaMemcpy(dev_wordList, &(wordList[0][0]), 20 * number_of_words * sizeof(char), cudaMemcpyHostToDevice);

    //Setup execution parameters
    int n_blocks = (number_of_words + 255)/256;
    int threads_per_block = 256;


    dim3 grid(n_blocks, 1, 1);
    dim3 threads(threads_per_block, 1, 1);


    cudaPrintfInit();
    testKernel<<<grid, threads>>>(dev_wordList);
    cudaDeviceSynchronize();
    cudaPrintfDisplay(stdout,true);
    cudaPrintfEnd();

(内核功能代码:)

__global__ void testKernel(char* d_wordList)
{
    //access thread id
    const unsigned int bid = blockIdx.x;
    const unsigned int tid = threadIdx.x;
    const unsigned int index = bid * blockDim.x + tid;

    cuPrintf("!! %c%c%c%c%c%c%c%c%c%c \n" , d_wordList[index * 20 + 0],
                                            d_wordList[index * 20 + 1],
                                            d_wordList[index * 20 + 2],
                                            d_wordList[index * 20 + 3],
                                            d_wordList[index * 20 + 4],
                                            d_wordList[index * 20 + 5],
                                            d_wordList[index * 20 + 6],
                                            d_wordList[index * 20 + 7],
                                            d_wordList[index * 20 + 8],
                                            d_wordList[index * 20 + 9]);
}

有没有办法更轻松地操纵它们? (我希望每个元素/位置都有一个单词)我尝试使用<string>,但我无法在CUDA设备代码中使用它们。

1 个答案:

答案 0 :(得分:1)

cuPrintf("%s\n", d_wordlist+(index*20));

应该有用吗? (假设你的字符串是零终止的)

更新

这一行:

char (*wordList)[text_length] = new char[10000][text_length];

对我来说很奇怪。通常,char的指针数组将按如下方式分配:

char** wordList = new char*[10000];
for (int i=0;i<10000;i++) wordList[i] = new char[20];

在这种情况下,wordList [i]将是指向字符串编号i的指针。

更新#2:

如果您需要将字符串存储为连续块,并且您确定没有任何字符串超过text_length + 1,那么您可以这样做:

char *wordList = new char[10000*text_length];

for(int i=0; i<number_of_words; i++)
     {
         file>>wordList+(i*text_length);
         cout<<wordList+(i*text_length)<<endl;
     }

在这种情况下,wordList +(i * text_length)将指向字符串编号i的开头,并且它将以0结尾,因为这是您从文件中读取它的方式,并且您将能够打印它用这个答案中指定的方式。但是,如果您的任何字符串长于text_length-1,您仍会遇到问题。