Skip to content

Commit 0235491

Browse files
authored
Merge pull request #455 from NVIDIA/bowa_fallback
fix: fix a typo for printing mini graph when debug
2 parents d03b0f9 + c823ebd commit 0235491

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

core/compiler.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,7 @@ torch::jit::script::Module CompileGraphWithFallback(const torch::jit::script::Mo
208208
for (auto& seg_block : segmented_blocks) {
209209
std::string cur_block_target =
210210
seg_block.target() == partitioning::SegmentedBlock::kTensorRT ? "TensorRT" : "Torch";
211-
LOG_INFO(*g << "(MiniGraphIn" << cur_block_target << "Block\n");
211+
LOG_INFO(*seg_block.g() << "(MiniGraphIn" << cur_block_target << "Block)\n");
212212
std::ostringstream trt_engine_id;
213213
trt_engine_id << reinterpret_cast<const int*>(&seg_block);
214214
if (seg_block.target() == partitioning::SegmentedBlock::kTensorRT) {

0 commit comments

Comments
 (0)