diff --git a/core/conversion/conversion.cpp b/core/conversion/conversion.cpp index c524e44245..23193df7e4 100644 --- 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 GetUnsupportedOpsInBlock(const torch::jit::Block* b ) { std::set 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");