From 3fbceb3a383fa69178c54592d1d49659c552fb6f Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Wed, 13 Jul 2022 13:18:55 +0800
Subject: [PATCH] fix(mgb/version): fix nvinfer.h not found

GitOrigin-RevId: 981dd9a7d2759b4d07a33ac451aacea0f1220070
---
 src/core/impl/version.cpp | 32 ++++++++++++++++++++++----------
 1 file changed, 22 insertions(+), 10 deletions(-)

diff --git a/src/core/impl/version.cpp b/src/core/impl/version.cpp
index 586854cf..1e547f97 100644
--- a/src/core/impl/version.cpp
+++ b/src/core/impl/version.cpp
@@ -26,27 +26,39 @@ Version mgb::get_version() {
 #endif
 }
 
-#if MGB_CUDA
+#if __has_include("NvInfer.h")
+#pragma GCC diagnostic push
+#pragma GCC diagnostic ignored "-Wnon-virtual-dtor"
+#pragma GCC diagnostic ignored "-Wdeprecated-declarations"
 #include "NvInfer.h"
+int mgb::get_tensorrt_version() {
+    return NV_TENSORRT_VERSION;
+}
+#pragma GCC diagnostic pop
+#else
+int mgb::get_tensorrt_version() {
+    return -1;
+}
+#endif
+
+#if __has_include("cuda.h")
 #include "cuda.h"
-#include "cudnn.h"
 int mgb::get_cuda_version() {
     return CUDA_VERSION;
 }
-int mgb::get_cudnn_version() {
-    return CUDNN_VERSION;
-}
-int mgb::get_tensorrt_version() {
-    return NV_TENSORRT_VERSION;
-}
 #else
 int mgb::get_cuda_version() {
     return -1;
 }
+#endif
+
+#if __has_include("cudnn.h")
+#include "cudnn.h"
 int mgb::get_cudnn_version() {
-    return -1;
+    return CUDNN_VERSION;
 }
-int mgb::get_tensorrt_version() {
+#else
+int mgb::get_cudnn_version() {
     return -1;
 }
 #endif