File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -487,18 +487,20 @@ bool VerifyConverterSupportForBlock(const torch::jit::Block* b) {
487
487
unsupported_msg << " https://www.github.com/nvidia/TRTorch/issues" << std::endl;
488
488
unsupported_msg << std::endl << " In Module:" << std::endl;
489
489
490
+ LOG_ERROR (unsupported_msg.str ());
491
+
490
492
for (const auto n : b->nodes ()) {
491
493
auto schema = n->maybeSchema ();
492
494
if (schema) {
493
495
for (const auto & x : unsupported_ops) {
494
496
if (x.first == schema->operator_name ()) {
495
- unsupported_msg << " Unsupported operator: " << *schema << std::endl;
496
- unsupported_msg << trtorch::core::util::GetPyTorchSourceCode (n) << std::endl;
497
+ LOG_ERROR (
498
+ " Unsupported operator: " << *schema << std::endl
499
+ << trtorch::core::util::GetPyTorchSourceCode (n) << std::endl);
497
500
}
498
501
}
499
502
}
500
503
}
501
- LOG_ERROR (unsupported_msg.str ());
502
504
return false ;
503
505
}
504
506
You can’t perform that action at this time.
0 commit comments