Skip to content

Commit

Permalink
fix(//core/conversion): Supress unnecessary debug messages
Browse files Browse the repository at this point in the history
Signed-off-by: Naren Dasan <naren@narendasan.com>
Signed-off-by: Naren Dasan <narens@nvidia.com>
  • Loading branch information
narendasan committed Jun 5, 2020
1 parent 09f5b13 commit 2b23874
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion core/conversion/conversion.cpp
Expand Up @@ -303,7 +303,7 @@ std::string ConvertBlockToEngine(const torch::jit::Block* b, ConversionInfo buil
std::set<std::string> GetUnsupportedOpsInBlock(const torch::jit::Block* b ) {
std::set<std::string> unsupported_ops;
for (const auto n : b->nodes()) {
if (!OpSupported(n) && n->kind() != torch::jit::prim::Loop) {
if (n->kind() != torch::jit::prim::Loop && !OpSupported(n)) {
auto schema = n->maybeSchema();
TRTORCH_CHECK(schema, "Unable to get schema for Node " << util::node_info(n) \
<< " (conversion.VerifyCoverterSupportForBlock");
Expand Down

0 comments on commit 2b23874

Please sign in to comment.