thrust与c++的STL进行向量copy时间比较

东方耀

//thrust与c++的STL进行向量copy时间比较


// c++ STL time(单位是微秒): 4514929
// thrust time: 547
//速度差距还是非常明显的

1. #include <thrust/device_vector.h>
   
2. #include <iostream>
   
3. #include <sys/time.h>
   
4. 
   
5. 
   
6. //thrust与c++的STL进行向量copy时间比较
   
7. 
   
8. // c++ STL time(单位是微秒): 4514929
   
9. // thrust time: 547
   
10. //速度差距还是非常明显的
    
11. 
    
12. 
    
13. int main(void){
    
14.     thrust::device_vector<int> D(1000000);
    
15.         thrust::sequence(D.begin(), D.end());
    
16. 
    
17.         std::vector<int> s(1000000);
    
18.     // 统计时间
    
19.     struct timeval startTime, endTime;
    
20. 
    
21.     gettimeofday(&startTime, NULL);
    
22.         std::copy(D.begin(), D.end(), s.begin());  // stl的copy方法
    
23.     gettimeofday(&endTime, NULL);
    
24.     printf("c++ STL time(单位是微秒): %ld\n",
    
25.     (endTime.tv_sec - startTime.tv_sec)*1000000 + (endTime.tv_usec - startTime.tv_usec));
    
26. 
    
27.     gettimeofday(&startTime, NULL);
    
28.         thrust::copy(D.begin(), D.end(), s.begin());   // thrust的copy方法
    
29.     gettimeofday(&endTime, NULL);
    
30.     printf("thrust time: %ld\n",
    
31.     (endTime.tv_sec - startTime.tv_sec)*1000000 + (endTime.tv_usec - startTime.tv_usec));
    
32. 
    
33.     return 0;
    
34. }
    
35. 
    
36. 
    

复制代码

zouqiqi

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

zouqiqi

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