提交 9e9e8ca0 编写于 作者: M Megvii Engine Team 提交者: Xu Xinran

fix(mgb): fix some compile errors

GitOrigin-RevId: ec38f8e15aeb8a5c6ba987194197f05cad39ee10
上级 ef267dac
......@@ -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 {
......
......@@ -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;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册