tensorRT---认识cuda-driver-api

zsffuture

已于 2022-05-07 19:39:27 修改

阅读量299

点赞数

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

于 2022-04-30 11:05:02 首次发布

本文链接：https://blog.csdn.net/weixin_42398658/article/details/124509120

版权

本文介绍了如何使用CUDA Driver API进行GPU上下文管理和CPU-GPU数据交换。主要内容包括CUcontext的创建与使用，以及通过cuMemAllocHost函数分配页锁定内存实现高效DMA数据传输。在实际操作中，需要注意数据类型匹配，如将整型数据转换为float类型以避免GPU解析错误。

摘要由CSDN通过智能技术生成

CUcontext

实战代码


// CUDA驱动头文件cuda.h
#include <cuda.h>   // include <> 和 "" 的区别    
#include <stdio.h>  // include <> : 标准库文件 
#include <string.h> // include "" : 自定义文件  详细情况请查看 readme.md -> 5

#define checkDriver(op)  __check_cuda_driver((op), #op, __FILE__, __LINE__)

bool __check_cuda_driver(CUresult code, const char* op, const char* file, int line){
    if(code != CUresult::CUDA_SUCCESS){    // 如果 成功获取CUDA情况下的返回值 与我们给定的值(0)不相等， 即条件成立， 返回值为flase
        const char* err_name = nullptr;    // 定义了一个字符串常量的空指针
        const char* err_message = nullptr;  
        cuGetErrorName(code, &err_name);    
        cuGetErrorString(code, &err_message);   
        printf("%s:%d  %s failed. \n  code = %s, message = %s\n", file, line, op, err_name, err_message); //打印错误信息
        return false;
    }
    return true;
}

int main(){

    // 检查cuda driver的初始化
    checkDriver(cuInit(0));

    // 为设备创建上下文
    CUcontext ctxA = nullptr;                                   // CUcontext 其实是 struct CUctx_st*（是一个指向结构体CUctx_st的指针）
    CUcontext ctxB &

最低0.47元/天解锁文章