亮點：通過cmake定義的選項在代碼中進行動態選擇
源碼:dler

1. 例子

代碼中通過宏　tx2 進行代碼的控制．
code

#ifndef NET_H #define NET_H#include <memory> #include <string> #include <vector> #include <iostream> #include <opencv2/opencv.hpp> #ifdef tx2 #include <NvInfer.h> #include <NvOnnxParser.h> #include <NvOnnxParserRuntime.h> #define num 0 #else #include <tensorrt/NvInfer.h> #include <tensorrt/NvOnnxParser.h> #include <tensorrt/NvOnnxParserRuntime.h> #define num 1 #endif#endif

我們在CMakeLists.txt中可以通過add_definitions(-Dtx2)來定義這個宏。但是好像add_definitions沒法通過cmake命令行來進行控制。所以這里在cmake中再聲明一個option進行選擇，完整的cmake為：

cmake_minimum_required(VERSION 3.2) project(test)option(ONTX2 "build project on tx2" OFF) IF(ONTX2)add_definitions(-Dtx2) ENDIF(ONTX2)set(source net.cpp) add_executable(test ${source})

編譯的時候通過以下兩種方式進行選擇

cmake -DONTX2=1 .. OR cmake -DONTX2=0 ..

2. 參考

• CMAKE 中　add_definitions的用法

總結

以上是生活随笔為你收集整理的【c++】——通过cmake定义宏add_definitions的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。