augustin64/tipe
1
1
Fork 0
mirror of https://github.com/augustin64/projet-tipe synced 2025-02-02 19:39:39 +01:00
Code Issues Packages Projects Releases Wiki Activity

matrix mul:Add comments

Browse Source
This commit is contained in:
augustin64 2023-05-25 10:38:45 +02:00
parent 46aa64da01
commit dd9613f159
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/cnn/matrix_multiplication.cu
View File

@ -9,24 +9,24 @@
#define BLOCKSIZE_y 16 #define BLOCKSIZE_y 16
#ifdef __CUDACC__ #ifdef __CUDACC__
__global__ void matrix_mul_kernel(float** Md, float** Nd, float** Pd, int n, int p, int q) { __global__ void matrix_mul_kernel(float** M, float** N, float** P, int n, int p, int q) {
// Chaque thread calcule toutes les multiplications utilisant l'élément Nd[tx][ty] // Ce fil calcule toutes les multiplications utilisant l'élément N[idx][idy]
int tx = (blockIdx.x*blockDim.x) + threadIdx.x; // Indice de colonne int idx = (blockIdx.x*blockDim.x) + threadIdx.x; // Indice de colonne
int ty = (blockIdx.y*blockDim.y) + threadIdx.y; // Indice de ligne int idy = (blockIdx.y*blockDim.y) + threadIdx.y; // Indice de ligne
if (tx >= p || ty >= q) { if (idx >= p || idy >= q) {
return; return; // On vérifie que l'on est bien à un emplacement valide
} }
for (int i = 0; i < n; i++) { for (int i = 0; i < n; i++) {
atomicAdd(&(Pd[i][ty]), Md[i][tx]*Nd[tx][ty]); atomicAdd(&(P[i][idy]), M[i][idx]*N[idx][idy]);
// P[i][ty] += P[i][tx] * N[tx][ty] // P[i][idy] += M[i][idx] * N[idx][idy]
} }
} }
void matrix_multiplication_device(float** m1, float** m2, float** result, int n, int p, int q) { void matrix_multiplication_device(float** m1, float** m2, float** result, int n, int p, int q) {
// Traitement // On découpe la tâche en un certain nombre de blocs,
// la taille d'un bloc étant limitée par CUDA à 1024
dim3 gridSize(i_div_up(p, BLOCKSIZE_x), i_div_up(q, BLOCKSIZE_y)); dim3 gridSize(i_div_up(p, BLOCKSIZE_x), i_div_up(q, BLOCKSIZE_y));
dim3 blockSize(BLOCKSIZE_x, BLOCKSIZE_y); dim3 blockSize(BLOCKSIZE_x, BLOCKSIZE_y);