tensorRT---认识cuda RuntimeAPI（认识thrust，认识cuda的错误机制）

zsffuture

已于 2022-05-21 10:41:35 修改

阅读量344

点赞数

分类专栏： c++ cuda 文章标签： c++ 算法

于 2022-05-01 20:40:57 首次发布

本文链接：https://blog.youkuaiyun.com/weixin_42398658/article/details/124531916

版权

本文深入探讨了在TensorRT中如何使用CUDA Runtime API，特别是介绍了Thrust库的运用，以及CUDA运行时的错误处理机制，帮助读者更好地理解和优化GPU计算。

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

thrust


#include <stdio.h>
#include <thrust/host_vector.h>
#include <thrust/device_vector.h>
#include <thrust/sort.h>
#include <iostream>
using namespace std;

__host__ __device__
int sort_func(int a, int b){
    return a > b;
}

int main(){

    int data[] = {5, 3, 1, 5, 2, 0};
    int ndata  = sizeof(data) / sizeof(data[0]);
    thrust::host_vector<int> array1(data, data + ndata);
    thrust::sort(array1.begin(), array1.end(), sort_func);

    thrust::device_vector<int> array2 = thrust::host_vector<int>(data, data + ndata);
    thrust::sort(array2.begin(), array2.end(), []__device__(int a, int b){return a < b;});

    printf("array1------------------------\n");
    for(int i = 0; i < array1.size(); ++i)
        cout << array1[i] << endl;

    printf("array2------------------------\n");
    for(int i = 0; i < array2.size(