diff --git a/core/conversion/conversion.cpp b/core/conversion/conversion.cpp index 27287fe568..9547ddb965 100644 --- a/core/conversion/conversion.cpp +++ b/core/conversion/conversion.cpp @@ -487,7 +487,8 @@ std::string ConvertBlockToEngine( std::unordered_map GetUnsupportedOpsInBlock(const torch::jit::Block* b) { std::unordered_map unsupported_ops; for (const auto n : b->nodes()) { - if (n->kind() != torch::jit::prim::Loop && n->kind() != torch::jit::prim::If && !OpSupported(n)) { + if (n->kind() != torch::jit::prim::Loop && n->kind() != torch::jit::prim::If && !OpSupported(n) && + n->kind() != torch::jit::prim::DictConstruct) { auto schema = n->maybeSchema(); TORCHTRT_CHECK( schema,