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 {
#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 {


+ 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:
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;


Loading…
Cancel
Save