Thrust快速入门教程（二）——Vector的使用

最新推荐文章于 2025-04-01 15:13:35 发布

忆幽梦

最新推荐文章于 2025-04-01 15:13:35 发布

阅读量8.3k

点赞数 2

分类专栏： CUDA 文章标签： vector include iostream 编译器 function 存储

本文链接：https://blog.csdn.net/dreampursue/article/details/6278737

版权

本文是Thrust库的快速入门教程，重点介绍了host_vector和device_vector的使用，包括初始化、元素访问、复制及resize操作。通过示例展示了如何在主机和设备间高效地管理数据，以及如何利用fill、copy和sequence函数进行数据操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Trust 提供了两个vector容器：host_vector 与 device_vector。按照命名规则，host_vector位于主机端，device_vector位于GPU设备端。Trust的vector容器与STL中的容器类似，是通用的容器，可以存储任何数据类型，可以动态调整大小。以下源代码展示如何使用Thrust的vector容器。

# include <thrust / host_vector .h> # include <thrust / device_vector .h> # include <iostream > int main ( void ) { // H has storage for 4 integers thrust :: host_vector <int > H (4); // initialize individual elements H [0] = 14; H [1] = 20; H [2] = 38; H [3] = 46; // H. size () returns the size of vector H std :: cout << "H has size " << H. size () << std :: endl ; // print contents of H for ( int i = 0; i < H. size (); i ++) std :: cout << "H[" << i << "] = " << H[i] << std :: endl ; // resize H H. resize (2) ; std :: cout << "H now has size " << H. size () << std :: endl ; // Copy host_vector H to device_vector D thrust :: device_vector <int > D =

最低0.47元/天解锁文章