Skip to content

Commit 01d98c7

Browse files
committed
feat: Enable prim::DictConstruct to fallback without conversion check error
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent 11bcb98 commit 01d98c7

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

Diff for: core/conversion/conversion.cpp

+2-1
Original file line numberDiff line numberDiff line change
@@ -487,7 +487,8 @@ std::string ConvertBlockToEngine(
487487
std::unordered_map<c10::OperatorName, std::string> GetUnsupportedOpsInBlock(const torch::jit::Block* b) {
488488
std::unordered_map<c10::OperatorName, std::string> unsupported_ops;
489489
for (const auto n : b->nodes()) {
490-
if (n->kind() != torch::jit::prim::Loop && n->kind() != torch::jit::prim::If && !OpSupported(n)) {
490+
if (n->kind() != torch::jit::prim::Loop && n->kind() != torch::jit::prim::If && !OpSupported(n) &&
491+
n->kind() != torch::jit::prim::DictConstruct) {
491492
auto schema = n->maybeSchema();
492493
TORCHTRT_CHECK(
493494
schema,

0 commit comments

Comments
 (0)