fix(//core/conversion): Supress unnecessary debug messages

Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Jun 5, 2020 · 2b23874 · 2b23874
1 parent 09f5b13
commit 2b23874
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/conversion/conversion.cpp b/core/conversion/conversion.cpp
@@ -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");