东方耀AI技术分享

标题: cuda编程：数组运算放GPU上操作,看占显存 [打印本页]

---

作者: 东方耀    时间: 2021-6-8 11:37
标题: 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    时间: 2021-11-23 19:40
让天下人人学会人工智能！人工智能的前景一片大好！

---

欢迎光临 东方耀AI技术分享 (http://www.ai111.vip/)

Powered by Discuz! X3.4