From 9e9e8ca0fe3864e642502a99ba961e0c5a5f99dc Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Wed, 15 Jul 2020 15:18:09 +0800
Subject: [PATCH] fix(mgb): fix some compile errors

GitOrigin-RevId: ec38f8e15aeb8a5c6ba987194197f05cad39ee10
---
 src/core/impl/comp_node/cuda/comp_node.cpp | 8 ++++++++
 src/tensorrt/impl/tensorrt_opr.cpp         | 2 ++
 2 files changed, 10 insertions(+)

diff --git a/src/core/impl/comp_node/cuda/comp_node.cpp b/src/core/impl/comp_node/cuda/comp_node.cpp
index 0ebd38be..467931ce 100644
--- a/src/core/impl/comp_node/cuda/comp_node.cpp
+++ b/src/core/impl/comp_node/cuda/comp_node.cpp
@@ -235,6 +235,7 @@ class CudaCompNode::CompNodeImpl final: public CompNode::Impl {
         }
 
         void add_callback(CudaHostFunc&& cb) override {
+#if CUDART_VERSION >= 10000
             activate();
             CudaHostFunc* func_ptr = new CudaHostFunc(std::move(cb));
             MGB_TRY {
@@ -244,6 +245,13 @@ class CudaCompNode::CompNodeImpl final: public CompNode::Impl {
                 delete func_ptr;
                 throw;
             });
+#else
+            MGB_MARK_USED_VAR(cb);
+            MGB_MARK_USED_VAR(cuda_host_func_caller);
+            mgb_throw(
+                    MegBrainError,
+                    "add_callback only support in cuda10.0 and later version");
+#endif
         }
 
         uint64_t get_uid() override {
diff --git a/src/tensorrt/impl/tensorrt_opr.cpp b/src/tensorrt/impl/tensorrt_opr.cpp
index 9e1a630a..95652025 100644
--- a/src/tensorrt/impl/tensorrt_opr.cpp
+++ b/src/tensorrt/impl/tensorrt_opr.cpp
@@ -83,9 +83,11 @@ void TensorRTOpr::Logger::log(nvinfer1::ILogger::Severity severity,
         case Severity::kINFO:
             mgb_log_debug("TRT_INFO: %s", msg);
             return;
+#if NV_TENSOR_RT_VERSION >= 6001
         case Severity::kVERBOSE:
             mgb_log_debug("TRT_VERBOSE: %s", msg);
             return;
+#endif
         default:
             mgb_log_debug("TRT_UNKNOWN: %s", msg);
             return;