From c823ebd3b262c3546cd758db28d9ed6cd2b855a3 Mon Sep 17 00:00:00 2001 From: Bo Wang Date: Wed, 12 May 2021 12:24:57 -0500 Subject: [PATCH] fix: fix a typo for debug Signed-off-by: Bo Wang --- core/compiler.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/core/compiler.cpp b/core/compiler.cpp index 17a6b50085..7f5e9e3076 100644 --- a/core/compiler.cpp +++ b/core/compiler.cpp @@ -208,7 +208,7 @@ torch::jit::script::Module CompileGraphWithFallback(const torch::jit::script::Mo for (auto& seg_block : segmented_blocks) { std::string cur_block_target = seg_block.target() == partitioning::SegmentedBlock::kTensorRT ? "TensorRT" : "Torch"; - LOG_INFO(*g << "(MiniGraphIn" << cur_block_target << "Block\n"); + LOG_INFO(*seg_block.g() << "(MiniGraphIn" << cur_block_target << "Block)\n"); std::ostringstream trt_engine_id; trt_engine_id << reinterpret_cast(&seg_block); if (seg_block.target() == partitioning::SegmentedBlock::kTensorRT) {