From 01d98c72bd4f8182d29de6ec9a9f638c78595a7a Mon Sep 17 00:00:00 2001
From: Dheeraj Peri <peri.dheeraj@gmail.com>
Date: Wed, 16 Mar 2022 23:19:14 -0700
Subject: [PATCH] feat: Enable prim::DictConstruct to fallback without
 conversion check error

Signed-off-by: Dheeraj Peri <peri.dheeraj@gmail.com>
---
 core/conversion/conversion.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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<c10::OperatorName, std::string> GetUnsupportedOpsInBlock(const torch::jit::Block* b) {
   std::unordered_map<c10::OperatorName, std::string> 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,