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

// check whether date is valid
var t = new Date('2011-07-07T11:20:00.000+00:00x');
valid = !isNaN(t.valueOf());
18
задан Gitmo 26 June 2009 в 04:38
поделиться

2 ответа

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

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

Например

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

Это создает 2D-массив размера 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;
}
16
ответ дан 30 November 2019 в 08:53
поделиться

Расплющить: сделать одномерным. Посмотрите, как это делается здесь

3
ответ дан 30 November 2019 в 08:53
поделиться
Другие вопросы по тегам:

Похожие вопросы: