From 4bf2a41dbac539f7d0f17aa127a223a70261a137 Mon Sep 17 00:00:00 2001 From: Naren Dasan Date: Thu, 22 Jul 2021 09:28:28 -0700 Subject: [PATCH] feat(//core/conversion): Adding error prefix to python source traceback Signed-off-by: Naren Dasan Signed-off-by: Naren Dasan --- core/conversion/conversion.cpp | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/core/conversion/conversion.cpp b/core/conversion/conversion.cpp index 17e554f21c..6d65cfeb92 100644 --- a/core/conversion/conversion.cpp +++ b/core/conversion/conversion.cpp @@ -487,18 +487,20 @@ bool VerifyConverterSupportForBlock(const torch::jit::Block* b) { unsupported_msg << "https://www.github.com/nvidia/TRTorch/issues" << std::endl; unsupported_msg << std::endl << "In Module:" << std::endl; + LOG_ERROR(unsupported_msg.str()); + for (const auto n : b->nodes()) { auto schema = n->maybeSchema(); if (schema) { for (const auto& x : unsupported_ops) { if (x.first == schema->operator_name()) { - unsupported_msg << " Unsupported operator: " << *schema << std::endl; - unsupported_msg << trtorch::core::util::GetPyTorchSourceCode(n) << std::endl; + LOG_ERROR( + "Unsupported operator: " << *schema << std::endl + << trtorch::core::util::GetPyTorchSourceCode(n) << std::endl); } } } } - LOG_ERROR(unsupported_msg.str()); return false; }