一、问题描述
项目中用到TensorRT进行模型的推理同时项目中用到了glog作为日志管理的工具。TensorRT中用到了显存开辟函数cudaMalloc等函数,因此加了CHECK函数校验是否开辟成功。由于glog中也有同名的CHECK函数。在编译阶段没有发现冲突,在执行过程中就总是出错。一开始没想到是CHECK校验不对,一致还以为是显存开辟失败,花了大约30分钟的时间来定位这个问题。
二、解决方案
加上命名空间来避免冲突
项目中用到TensorRT进行模型的推理同时项目中用到了glog作为日志管理的工具。TensorRT中用到了显存开辟函数cudaMalloc等函数,因此加了CHECK函数校验是否开辟成功。由于glog中也有同名的CHECK函数。在编译阶段没有发现冲突,在执行过程中就总是出错。一开始没想到是CHECK校验不对,一致还以为是显存开辟失败,花了大约30分钟的时间来定位这个问题。
加上命名空间来避免冲突
本文标题:glog中的CHECK和cuda中的CHECK冲突
本文链接:https://www.haomeiwen.com/subject/owdizktx.html
网友评论