Browse Source

fix(mgb): fix some compile errors

GitOrigin-RevId: ec38f8e15a
tags/v1.0.0-rc1
Megvii Engine Team Xu Xinran 4 years ago
parent
commit
9e9e8ca0fe
2 changed files with 10 additions and 0 deletions
  1. +8
    -0
      src/core/impl/comp_node/cuda/comp_node.cpp
  2. +2
    -0
      src/tensorrt/impl/tensorrt_opr.cpp

+ 8
- 0
src/core/impl/comp_node/cuda/comp_node.cpp View File

@@ -235,6 +235,7 @@ class CudaCompNode::CompNodeImpl final: public CompNode::Impl {
} }


void add_callback(CudaHostFunc&& cb) override { void add_callback(CudaHostFunc&& cb) override {
#if CUDART_VERSION >= 10000
activate(); activate();
CudaHostFunc* func_ptr = new CudaHostFunc(std::move(cb)); CudaHostFunc* func_ptr = new CudaHostFunc(std::move(cb));
MGB_TRY { MGB_TRY {
@@ -244,6 +245,13 @@ class CudaCompNode::CompNodeImpl final: public CompNode::Impl {
delete func_ptr; delete func_ptr;
throw; 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 { uint64_t get_uid() override {


+ 2
- 0
src/tensorrt/impl/tensorrt_opr.cpp View File

@@ -83,9 +83,11 @@ void TensorRTOpr::Logger::log(nvinfer1::ILogger::Severity severity,
case Severity::kINFO: case Severity::kINFO:
mgb_log_debug("TRT_INFO: %s", msg); mgb_log_debug("TRT_INFO: %s", msg);
return; return;
#if NV_TENSOR_RT_VERSION >= 6001
case Severity::kVERBOSE: case Severity::kVERBOSE:
mgb_log_debug("TRT_VERBOSE: %s", msg); mgb_log_debug("TRT_VERBOSE: %s", msg);
return; return;
#endif
default: default:
mgb_log_debug("TRT_UNKNOWN: %s", msg); mgb_log_debug("TRT_UNKNOWN: %s", msg);
return; return;


Loading…
Cancel
Save