fix: fix a typo for debug

Signed-off-by: Bo Wang <[email protected]>
pytorch · May 12, 2021 · c823ebd · c823ebd
1 parent 6e0da4d
commit c823ebd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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<const int*>(&seg_block);
         if (seg_block.target() == partitioning::SegmentedBlock::kTensorRT) {