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.
1 parent 6e0da4d commit c823ebdCopy full SHA for c823ebd
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