Sortie du programme C/CUDA

Question

Ce qui suit est un exemple de programmation CUDA qui est fondamentalement C mais avec des fonctions NVIDIA CUDA à l'intérieur. J'ai essayé d'interpréter cet exemple de code et de comprendre ce qu'il essaie de faire. Ma question est la suivante: le programme compile-t-il très bien, mais quels sont les arguments? Par exemple, ce programme CUDA est en cours d'exécution dans un émulateur Linux mais sur l'exécution ./program il retourne:

Utilisation: Numéro ./program Segmentation fault

Quels sont les arguments d'entrée des programmes. Je vous remercie.

#include <assert.h> 
#include <stdio.h> 

//#define N 100000 

__host__ void saxpy_host(int length, float alpha, float * x, float * y) 
{ 
    for (int i = 0; i < length; ++i) 
    y[i] = alpha*x[i] + y[i]; 
} 

__global__ void saxpy (int length, float alpha, float * x, float * y) 
{ 
    int i; 
    i = blockIdx.x*blockDim.x + threadIdx.x; 
    if (i < length) y[i] = alpha*x[i]+y[i]; 
    __syncthreads(); 
} 

int main(int argc, char* argv[]) { 

if (argc != 2) { 
    printf("Usage: %s number\n", argv[0]); 
    return -1; 
} 

int N = atoi(argv[1]); 

// host data 
float alpha = 0.5; 
float x[N], xback[N]; 
float y[N], yback[N]; 

int size; 
int i; 
int blocks; 

// determining size 
size = sizeof(float)*N; 


// device data 
float * dxp, * dyp; 


// fill host data 
for (i = 0; i < N; i++) { 
    x[i] = (float) (rand() % 128); 
    y[i] = (float) (rand() % 256); 
} 


// Allocating and Moving data to device 
cudaMalloc((void**) &dxp, size); 
cudaMalloc((void**) &dyp, size); 
cudaMemcpy (dxp, x, size, cudaMemcpyHostToDevice); 
cudaMemcpy (dyp, y, size, cudaMemcpyHostToDevice); 

// size of thread blocks 
blocks = (N + 31)/32; 
saxpy <<< blocks, 32 >>> (N, alpha, dxp, dyp); 

// bring back data 
cudaMemcpy (xback, dxp, size, cudaMemcpyDeviceToHost); 
cudaMemcpy (yback, dyp, size, cudaMemcpyDeviceToHost); 

// Calculating host SAXPY 
saxpy_host (N, alpha, (float *) &x, (float *) &y); 

// checking computation on host matches computation on GPU 
for (i = 0; i < N; i++) { 
    assert (yback[i] == y[i]) ; 
    //printf ("%i %f %f \n", i, yback[i], y[i]); 
} 


// free device data 
cudaFree(dxp); cudaFree(dyp); 

return 0; 
} 

Answer 1

int N = atoi(argv[1]); 

le programme prend un entier comme argument de ligne de commande. (Essayez de l'appeler ./program 5, par exemple.)

Il calcule ensuite un SAXPY (Un ancien terme provenant des premières implémentations de BLAS, mais il a collé. Cela signifie "unique (précision, alias flottant) réel alpha x plus y" .) avec des vecteurs de dimension N.