Asignación de matriz 2D en la memoria del dispositivo en CUDA

Question

¿Cómo puedo asignar y transferir (hacia y desde el host) matrices 2D en la memoria del dispositivo en Cuda?

Answer 1

Encontré una solución a este problema. No tuve que aplanar la matriz.

La función incorporada cudaMallocPitch() hizo el trabajo. Y podría transferir la matriz hacia y desde el dispositivo usando la función cudaMemcpy2D().

Por ejemplo

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

Esto crea una matriz 2D de tamaño a * b con el terreno de juego como pasa como parámetro.

El siguiente código crea una matriz 2D y recorre los elementos. Se compila fácilmente, puede usarlo.

#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; 
} 

Answer 2

Acoplarlo: hacerlo unidimensional. Vea cómo se hace here

Answer 3

El código de su dispositivo podría ser más rápido. Intente utilizar los hilos más.

__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]; 
    } 
} 

Luego se calcula la asignación de bloques y subprocesos adecuada para que cada subproceso trate con un solo elemento.