ODLA: add version check API

Author: Weiming Zhao

ODLA/include/ODLA/odla_version.h

@@ -33,6 +33,12 @@ extern "C" {
 //! \brief ODLA version number.
 #define ODLA_VERSION_NUMBER ((ODLA_MAJOR)*100 + (ODLA_MINOR)*10 + (OLDA_PATCH))
 
+//! \brief Get version info of runtime library.
+/*!
+  \return NULL-terminated string of version info.
+*/
+const char* odla_GetVersionString();
+
 #ifdef __cplusplus
 } // C extern
 #endif

ODLA/platforms/include/odla_impl_common.h

@@ -0,0 +1,43 @@
+//===- odla_impl_common.h -------------------------------------------------===//
+//
+// Copyright (C) 2019-2022 Alibaba Group Holding Limited.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+// =============================================================================
+
+#ifndef _ODLA_PLATFORMS_ODLA_COMMON_H_
+#define _ODLA_PLATFORMS_ODLA_COMMON_H_
+
+#include "version.inc"
+
+#define STR_HELPER(x) #x
+#define STR(x) STR_HELPER(x)
+
+#define VERSION_STR(major, minor, patch, build) \
+  STR(major) "." STR(minor) "." STR(patch) "." STR(build)
+
+#ifndef NDEBUG
+#define ODLA_BUILD_TYPE "DEBUG"
+#else
+#define ODLA_BUILD_TYPE "RELEASE"
+#endif
+
+#define ODLA_VERSION_COMMON_INFO                                        \
+  "Repo: " ODLA_REPOSITORY ", Rev:" ODLA_REVISION "\nOS:" HALO_BUILT_OS \
+  "\nBuild Type: " ODLA_BUILD_TYPE "\nBuilt on: " __DATE__ " " __TIME__
+
+#define ODLA_VERSION_STR(lib_name, major, minor, patch, build, extra_info) \
+  "ODLA library name: " lib_name "\nVersion: " VERSION_STR(                \
+      major, minor, patch, build) "\n" ODLA_VERSION_COMMON_INFO            \
+                                  "\n" extra_info
+#endif

ODLA/platforms/tensorrt/CMakeLists.txt

@@ -37,7 +37,7 @@ add_subdirectory(plugins)
 add_odla_library(odla_tensorrt SHARED odla_tensorrt.cc)
 add_dependencies(odla_tensorrt ODLA_TRTPLUGIN)
 
-target_include_directories(odla_tensorrt PRIVATE ${CUDA_ROOT}/include)
+target_include_directories(odla_tensorrt PRIVATE ${CUDA_ROOT}/include ${CMAKE_BINARY_DIR}/include/halo)
 target_compile_options(odla_tensorrt PRIVATE -Wno-deprecated-declarations)
 target_link_libraries(odla_tensorrt ODLA ${nvinfer} ${CUDART_LIB} ${nvinfer_plugin})
 set_target_properties(odla_tensorrt PROPERTIES LINK_FLAGS

ODLA/platforms/tensorrt/odla_tensorrt.cc

@@ -36,7 +36,7 @@
 #include <unordered_set>
 #include <vector>
 
-#include "ODLA/odla_common.h"
+#include "../include/odla_impl_common.h"
 #include "common.h"
 #include "plugins/initPlugin.h"
 
@@ -46,6 +46,22 @@ using namespace nvinfer1;
 #error This library requires minimum ODLA version 0.5
 #endif
 
+#define ODLA_TRT_MAJOR HALO_VERSION_MAJOR
+#define ODLA_TRT_MINOR HALO_VERSION_MINOR
+#define ODLA_TRT_PATCH HALO_VERSION_PATCH
+#define ODLA_TRT_BUILD 0
+
+const char* odla_GetVersionString() {
+#define EXTRA_INFO                                                          \
+  "Built with TensorRT: " VERSION_STR(NV_TENSORRT_MAJOR, NV_TENSORRT_MINOR, \
+                                      NV_TENSORRT_PATCH, NV_TENSORRT_BUILD)
+  const char* version =
+      ODLA_VERSION_STR("ODLA for TensorRT", ODLA_TRT_MAJOR, ODLA_TRT_MINOR,
+                       ODLA_TRT_PATCH, ODLA_TRT_BUILD, EXTRA_INFO);
+#undef EXTRA_INFO
+  return version;
+}
+
 // Explicitly load cuda runtime before all other ctors, so cuda rt will be
 // released after calling dtors of all other global objs. This avoids the error
 // of "driver shutting down".