glog中的CHECK和cuda中的CHECK冲突

一、问题描述

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

二、解决方案

加上命名空间来避免冲突

你可能感兴趣的:(glog中的CHECK和cuda中的CHECK冲突)