thrust/functional.h中的算子：negate、modulus

东方耀

thrust/functional.h中的算子：negate、modulus

1. #include <thrust/device_vector.h>
   
2. #include <thrust/transform.h>
   
3. #include <thrust/sequence.h>
   
4. #include <thrust/copy.h>
   
5. #include <thrust/fill.h>
   
6. #include <thrust/replace.h>
   
7. #include <thrust/functional.h>
   
8. #include <iostream>
   
9. 
   
10. 
    
11. //Thrust提供了大量常见的并行算法。其中许多算法在C ++标准库中都有直接类比
    
12. //Thrust中的所有算法都具有主机和设备的实现
    
13. //除了thrust::copy可以在主机和设备之间复制数据之外，
    
14. //Thrust算法的所有迭代器参数都应该位于同一位置：要么全部在主机上，要么全部在设备上。违反此要求时，编译器将生成错误消息
    
15. //thrust::fill，它将范围的所有元素设置为指定值。其他转变包括thrust::sequence，thrust::replace当然thrust::transform
    
16. 
    
17. //转换是将操作应用于一组（零个或多个）输入范围中的每个元素，然后将结果存储在目标范围内的算法
    
18. 
    
19. 
    
20. //thrust/functional.h中的算子：negate、modulus
    
21. 
    
22. 
    
23. int main(void)
    
24. {
    
25.   // allocate three device_vectors with 10 elements
    
26.   thrust::device_vector<int> X(10);
    
27.   thrust::device_vector<int> Y(10);
    
28.   thrust::device_vector<int> Z(10);
    
29.   //文件中thrust/functional.h:negate、modulus
    
30. 
    
31.   // initialize X to 0,1,2,3, ....
    
32.   thrust::sequence(X.begin(), X.end());
    
33. 
    
34.   // compute Y = -X  negate负数
    
35.   thrust::transform(X.begin(), X.end(), Y.begin(), thrust::negate<int>());
    
36. 
    
37.   // fill Z with twos
    
38.   thrust::fill(Z.begin(), Z.end(), 2);
    
39. 
    
40.   // compute Y = X mod 2
    
41.   thrust::transform(X.begin(), X.end(), Z.begin(), Y.begin(), thrust::modulus<int>());
    
42. 
    
43.   // replace all the ones in Y with tens
    
44.   thrust::replace(Y.begin(), Y.end(), 1, 10);
    
45. 
    
46.   // print Y  拷贝还可以这样用啊？
    
47.   thrust::copy(Y.begin(), Y.end(), std::ostream_iterator<int>(std::cout, "\n"));
    
48.    
    
49.   return 0;   
    
50. }
    
51. 
    
52. 
    
53. 
    

复制代码