cuda并行编程：原子操作atomic之计算直方图,比较了cuda与cpu...

东方耀

cuda并行编程：原子操作atomic之计算直方图,比较了cuda与cpu的时间

1. #include <stdio.h>
   
2. #include <sys/time.h>
   
3. 
   
4. 
   
5. __global__ void get_hist(float *a, int *hist)
   
6. {
   
7.         //块数目 线程数目
   
8.         int tid = threadIdx.x;
   
9.         int bid = blockIdx.x;
   
10.         int idx = tid + bid * blockDim.x;
    
11.         //原子操作 加 cuda自带的api  一种锁的机制
    
12.         atomicAdd(&hist[(int)a[idx]], 1);
    
13. }
    
14. 
    
15. //cuda并行编程：原子操作atomic之计算直方图,比较了cuda与cpu的时间
    
16. int main()
    
17. {
    
18.         int size = 32000000;
    
19.         float *a = new float[size];
    
20. 
    
21.         int length = 10;
    
22.         for(int i = 0; i < size; i++)
    
23.         {
    
24.                 a[i] = i*(i+1) % length;
    
25.         }
    
26. 
    
27.         int hist[length] = {0};
    
28. 
    
29.         float *aGpu;
    
30.         cudaMalloc(&aGpu, size * sizeof(float));
    
31.         cudaMemcpy(aGpu, a, size * sizeof(float), cudaMemcpyHostToDevice);
    
32. 
    
33.         int *histGpu;
    
34.         cudaMalloc(&histGpu, length * sizeof(int));
    
35.         cudaMemcpy(histGpu, hist, length * sizeof(int), cudaMemcpyHostToDevice);
    
36.         // 统计时间
    
37.         struct timeval startTime, endTime;
    
38.         gettimeofday(&startTime, NULL);
    
39.         // get_hist<<<1, size>> >(aGpu, histGpu);
    
40.         get_hist<<<size / 512, 512>> >(aGpu, histGpu);
    
41.         cudaMemcpy(hist, histGpu, length * sizeof(int), cudaMemcpyDeviceToHost);  //这个比较耗时 相比GPU运算
    
42.         gettimeofday(&endTime, NULL);
    
43.         printf("gpu计算的结果(次数统计)：\n");
    
44. 
    
45.         for(int i=0;i<10;i++){
    
46.                 printf("%d,", hist[i]);
    
47.         }
    
48.         printf("cuda use time(单位是微秒): %ld\n",
    
49.                 (endTime.tv_sec - startTime.tv_sec)*1000000 + (endTime.tv_usec - startTime.tv_usec));
    
50.         //数据清0
    
51.         memset(hist, 0, sizeof(hist));
    
52.         gettimeofday(&startTime, NULL);
    
53.         for(int i = 0; i < size; i++)
    
54.         {
    
55.                 hist[(int)a[i]] += 1;
    
56.         }
    
57.         gettimeofday(&endTime, NULL);
    
58.         printf("cpu计算的结果(次数统计)：\n");
    
59.         for(int i=0;i<10;i++){
    
60.                 printf("%d,", hist[i]);
    
61.         }
    
62.         printf("cpu use time: %ld\n",
    
63.                 (endTime.tv_sec - startTime.tv_sec)*1000000 + (endTime.tv_usec - startTime.tv_usec));
    
64.        
    
65. 
    
66.         return 0;
    
67. }

复制代码

zouqiqi

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