美文网首页
glog中的CHECK和cuda中的CHECK冲突

glog中的CHECK和cuda中的CHECK冲突

作者: RunningJiang | 来源:发表于2021-01-25 17:27 被阅读0次

    一、问题描述

    项目中用到TensorRT进行模型的推理同时项目中用到了glog作为日志管理的工具。TensorRT中用到了显存开辟函数cudaMalloc等函数,因此加了CHECK函数校验是否开辟成功。由于glog中也有同名的CHECK函数。在编译阶段没有发现冲突,在执行过程中就总是出错。一开始没想到是CHECK校验不对,一致还以为是显存开辟失败,花了大约30分钟的时间来定位这个问题。

    二、解决方案

    加上命名空间来避免冲突

    相关文章

      网友评论

          本文标题:glog中的CHECK和cuda中的CHECK冲突

          本文链接:https://www.haomeiwen.com/subject/owdizktx.html