-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathdot_prod.cu
More file actions
78 lines (57 loc) · 1.81 KB
/
Copy pathdot_prod.cu
File metadata and controls
78 lines (57 loc) · 1.81 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
#include <stdio.h>
#include <stdlib.h>
#define N 256*128
#define THREADS_PER_BLOCK 128
#define N_BLOCKS N/THREADS_PER_BLOCK
// Kernel for dot product
__global__ void dot( int *a, int *b, int *c ) {
__shared__ int prod[THREADS_PER_BLOCK]; // Shared memory
int index = blockIdx.x * blockDim.x + threadIdx.x;
prod[threadIdx.x] = a[index] * b[index];
__syncthreads(); // Threads synchronization
if( threadIdx.x == 0) {
int par_sum = 0;
for(int i=0; i<THREADS_PER_BLOCK; i++)
par_sum += prod[threadIdx.x]; // Threads reduction
atomicAdd(c,par_sum); // Blocks reduction
}
}
// Main program
int main(void){
int *a,*b,*c; // Host copies
int *a_dev,*b_dev,*c_dev; // Device copies
int size = N*sizeof(int); // Size of N integer
// Allocate host memory
a = (int *) malloc (size);
b = (int *) malloc (size);
c = (int *) malloc (sizeof(int));
// Allocate device memory
cudaMalloc( (void**)&a_dev, size);
cudaMalloc( (void**)&b_dev, size);
cudaMalloc( (void**)&c_dev, sizeof(int));
// Initialize
for (int i=0; i<N; i++){
a[i] = 1;
b[i] = 1;
}
*c = 0;
// Copy inputs to device
cudaMemcpy( a_dev, a, size , cudaMemcpyHostToDevice );
cudaMemcpy( b_dev, b, size , cudaMemcpyHostToDevice );
cudaMemcpy( c_dev, c, sizeof(int), cudaMemcpyHostToDevice );
// Launch kernel on device
dot <<< N_BLOCKS , THREADS_PER_BLOCK >>> (a_dev, b_dev, c_dev);
// Copy device result back to host
cudaMemcpy( c, c_dev, sizeof(int), cudaMemcpyDeviceToHost );
// Print result
printf("%d\n",*c);
// Free device memory
cudaFree(a_dev);
cudaFree(b_dev);
cudaFree(c_dev);
// Free host memory
free(a);
free(b);
free(c);
return 0;
}