We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents d03b0f9 + c823ebd commit 0235491Copy full SHA for 0235491
core/compiler.cpp
@@ -208,7 +208,7 @@ torch::jit::script::Module CompileGraphWithFallback(const torch::jit::script::Mo
208
for (auto& seg_block : segmented_blocks) {
209
std::string cur_block_target =
210
seg_block.target() == partitioning::SegmentedBlock::kTensorRT ? "TensorRT" : "Torch";
211
- LOG_INFO(*g << "(MiniGraphIn" << cur_block_target << "Block\n");
+ LOG_INFO(*seg_block.g() << "(MiniGraphIn" << cur_block_target << "Block)\n");
212
std::ostringstream trt_engine_id;
213
trt_engine_id << reinterpret_cast<const int*>(&seg_block);
214
if (seg_block.target() == partitioning::SegmentedBlock::kTensorRT) {
0 commit comments