cuda编程：数组运算放GPU上操作,看占显存

东方耀

cuda编程：数组运算放GPU上操作,看占显存

1. #include <stdio.h>
   
2. #include <unistd.h>
   
3. 
   
4. __global__ void add(int *a, int *b, int *c, int num){
   
5.     int i = threadIdx.x;
   
6.     //线程id 每个线程是并行执行
   
7.     //如果数据量很大，就需要很多block和thread；当足够大的数据时，需要考虑数据的划分的！
   
8.     int j = blockIdx.x;
   
9.     printf("线程id=%d;block块id=%d\n", i, j);
   
10.     if (i<num){
    
11.         //防止溢出
    
12.         c[i] = a[i] + b[i];
    
13.     }
    
14.    
    
15. }
    
16. //cuda编程：数组运算放GPU上操作,看占显存
    
17. int main(){
    
18.     int num = 30;
    
19.     int a[num], b[num], c[num];
    
20.     int *a_gpu, *b_gpu, *c_gpu;
    
21. 
    
22.     //cudaSetDevice(0);
    
23.     for(int i=0;i<num;i++){
    
24.         a[i] = i;
    
25.         b[i] = i*i;
    
26.     }
    
27.     cudaMalloc(&a_gpu, num*sizeof(int));
    
28.     cudaMalloc(&b_gpu, num*sizeof(int));
    
29.     cudaMalloc(&c_gpu, num*sizeof(int));
    
30. 
    
31.     cudaMemcpy(a_gpu, a, num * sizeof(int), cudaMemcpyHostToDevice);
    
32.     cudaMemcpy(b_gpu, b, num * sizeof(int), cudaMemcpyHostToDevice);
    
33. 
    
34.     // 具体计算操作
    
35.     add<<<1,num>>>(a_gpu,b_gpu,c_gpu,num);
    
36. 
    
37.     cudaMemcpy(c, c_gpu, num*sizeof(int), cudaMemcpyDeviceToHost);
    
38. 
    
39.    
    
40. 
    
41.     //看看结果
    
42. 
    
43.     for (int i=0;i<num;i++){
    
44.         printf("看结果:%d+%d=%d\n", a[i],b[i],c[i]);
    
45.     }
    
46. 
    
47.     usleep(10*1000*1000);  //10秒 看显存到底占用多少M?
    
48. 
    
49.     cudaFree(a_gpu);
    
50.     cudaFree(b_gpu);
    
51.     cudaFree(c_gpu);
    
52. 
    
53.     return 0;
    
54. }
    

复制代码

zouqiqi

让天下人人学会人工智能！人工智能的前景一片大好！