Выделить 2D массив в памяти устройства в CUDA - PullRequest
Новая
       20

Выделить 2D массив в памяти устройства в CUDA

18 голосов
/ 26 июня 2009

Как разместить и передать (на хост и с хоста) 2D-массивы в памяти устройства в Cuda?

Ответы [ 3 ]

16 голосов
/ 26 июня 2009

Я нашел решение этой проблемы. Мне не нужно было выравнивать массив.

Встроенная функция cudaMallocPitch() сделала свою работу. И я мог бы передать массив на устройство и с него, используя функцию cudaMemcpy2D().

Например 

cudaMallocPitch((void**) &array, &pitch, a*sizeof(float), b);

Это создает двумерный массив размером a * b с шагом, передаваемым в качестве параметра.

Следующий код создает двумерный массив и зацикливает элементы. Он легко компилируется, вы можете использовать его. 

#include<stdio.h>
#include<cuda.h>
#define height 50
#define width 50

// Device code
__global__ void kernel(float* devPtr, int pitch)
{
    for (int r = 0; r < height; ++r) {
        float* row = (float*)((char*)devPtr + r * pitch);
        for (int c = 0; c < width; ++c) {
             float element = row[c];
        }
    }
}

//Host Code
int main()
{

float* devPtr;
size_t pitch;
cudaMallocPitch((void**)&devPtr, &pitch, width * sizeof(float), height);
kernel<<<100, 512>>>(devPtr, pitch);
return 0;
}
3 голосов
/ 26 июня 2009

Сгладьте его: сделайте его одномерным. Посмотрите, как это делается здесь

2 голосов
/ 31 июля 2012

Код вашего устройства может быть быстрее. Попробуйте использовать темы больше. 

__global__ void kernel(float* devPtr, int pitch)
{
    int r = threadIdx.x;

    float* row = (float*)((char*)devPtr + r * pitch);
    for (int c = 0; c < width; ++c) {
         float element = row[c];
    }
}

Затем вы вычисляете распределение блоков и потоков так, чтобы каждый поток имел дело с одним элементом.

...