Как разместить структуру на GPU в CUDA - PullRequest
Новая
       97

Как разместить структуру на GPU в CUDA

1 голос
/ 07 апреля 2020

Я пытаюсь скопировать структуру в графический процессор, отредактировать ее в графическом процессоре, а затем вернуть в ЦПУ и использовать. Это моя структура 

typedef struct COOMatrix {
unsigned int numRows;
unsigned int numCols;
unsigned int nnz;
unsigned int capacity;
unsigned int* rowIdxs;
unsigned int* colIdxs;
float* values;
} COOMatrix;

Ниже приведена упрощенная версия моего кода 

 __global__ void spmspm(COOMatrix *result){ 
    result->rowIdxs[0] = 1;
    result->colIdxs[0] = 1;
    result->values[0] = 5;
}
void sparseNN(Vector* result, COOMatrix* outBuffer, COOMatrix** layerWeights, float bias, unsigned int numLayers) {

//outBuffer_d allocation
COOMatrix *outBuffer_d;
unsigned int* out_rowIdxs_d;
unsigned int* out_colIdxs_d;
float* out_values_d;
cudaMalloc((void**)&outBuffer_d, sizeof(COOMatrix));
cudaMalloc((void**)&out_rowIdxs_d, outBuffer->capacity * sizeof(unsigned int));
cudaMalloc((void**)&out_colIdxs_d, outBuffer->capacity * sizeof(unsigned int));
cudaMalloc((void**)&out_values_d, outBuffer->capacity * sizeof(float));



//copying outbuffer
cudaMemcpy(outBuffer_d, outBuffer, sizeof(COOMatrix), cudaMemcpyHostToDevice);
cudaMemcpy(out_rowIdxs_d, outBuffer->rowIdxs, outBuffer->capacity * sizeof(unsigned int), cudaMemcpyHostToDevice);
cudaMemcpy(out_colIdxs_d, outBuffer->colIdxs, outBuffer->capacity * sizeof(unsigned int), cudaMemcpyHostToDevice);
cudaMemcpy(out_values_d, outBuffer->values, outBuffer->capacity * sizeof(float), cudaMemcpyHostToDevice);
cudaMemcpy(&(outBuffer_d->rowIdxs), &out_rowIdxs_d, sizeof(unsigned int*), cudaMemcpyHostToDevice);
cudaMemcpy(&(outBuffer_d->colIdxs), &out_colIdxs_d, sizeof(unsigned int*), cudaMemcpyHostToDevice);
cudaMemcpy(&(outBuffer_d->values), &out_values_d, sizeof(float*), cudaMemcpyHostToDevice);
cudaDeviceSynchronize();


spmspm <<<1, 1>>> (outBuffer_d);


//copy back       
cudaMemcpy(outBuffer->rowIdxs, outBuffer_d->rowIdxs, outBuffer->capacity * sizeof(unsigned int), cudaMemcpyDeviceToHost);
cudaMemcpy(outBuffer->colIdxs, outBuffer_d->colIdxs, outBuffer->capacity * sizeof(unsigned int), cudaMemcpyDeviceToHost);
cudaMemcpy(outBuffer->values, outBuffer_d->values, outBuffer->capacity * sizeof(float), cudaMemcpyDeviceToHost);

printf("%f \n", outBuffer->values[0]);

}

Это приводит к ошибке сегментации. Я не понимаю, где я ошибся. Кто-нибудь может помочь?

...