Thrust快速入门教程（三）——迭代器与静态调度

本文主要是介绍Thrust快速入门教程（三）——迭代器与静态调度，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

http://blog.csdn.net/dreampursue/article/details/6278752

在这节中我们曾使用了这样的表达式，H.begin() 、H.end()、D.begin() + 7。begin()与end()的返回值在C++中被称为迭代器。vector的迭代器类似于数组的指针，用于指向数组的某个元素。H.begin()是指向H容器中数组第一个元素的迭代器。类似，H.end()指向H容器中的最后一个元素。

虽然说迭代器类似于指针，但它有着更丰富的作用。可以注意到在使用thrust::fill的时候我们并不需要指明这是device_vector的迭代器。这些信息包含在了D.begin()的返回值的迭代器类型中，其类型不同于H.begin()的返回值。当Thrust中的函数调用时，将根据迭代器的类型选择使用主机端还是设备端的算法实现。因为主机/设备调度是在编译时解析，所以这一过程被称为被称为静态调度。请注意，这意味着运行时没有调度进程的开销。

你可能想知道当“原始”指针作为Thrust函数的参数会如何。和STL一样，Thrust允许这种用法，它会调度主机端算法。如果传入的指针是指向设备端内存的指针，那么在调用函数之前需要用thrust::device_ptr封装。例如：

[cpp]  view plain copy  
 size_t N = 10;  
 // raw pointer to device memory  
 int * raw_ptr ;  
 cudaMalloc (( void **) & raw_ptr , N * sizeof ( int ));  
 // wrap raw pointer with a device_ptr  
 thrust :: device_ptr <int > dev_ptr ( raw_ptr );  
 // use device_ptr in thrust algorithms  
 thrust :: fill ( dev_ptr , dev_ptr + N, ( int ) 0);  

如需从 device_ptr中提取“原始”指针需要使用raw_pointer_cast，用法如下：

[cpp]  view plain copy  
 size_t N = 10;  
 // create a device_ptr  
 thrust :: device_ptr <int > dev_ptr = thrust :: device_malloc <int >(N);  
 // extract raw pointer from device_ptr  
 int * raw_ptr = thrust :: raw_pointer_cast ( dev_ptr );  

迭代器另一个区别于指针的地方在于它可以遍历各种数据结构。例如，STL提供了链表容器（std::list），提供双向的（但不是随机访问）的迭代器。虽然Thrust不提供这类容器的设备端实现，但是与它们兼容。

[cpp]  view plain copy  
 # include <thrust / device_vector .h>  
 # include <thrust / copy .h>  
 # include <list >  
 # include <vector >  
 int main ( void )  
 {  
 // create an STL list with 4 values  
 std :: list <int > stl_list ;  
 stl_list . push_back (10) ;  
 stl_list . push_back (20) ;  
 stl_list . push_back (30) ;  
 stl_list . push_back (40) ;  
 // initialize a device_vector with the list  
 thrust :: device_vector <int > D( stl_list . begin () , stl_list . end ());  
 // copy a device_vector into an STL vector  
 std :: vector <int > stl_vector (D. size ());  
 thrust :: copy (D. begin () , D. end () , stl_vector . begin ());  
 return 0;  
 }  