We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 11bcb98 commit 01d98c7Copy full SHA for 01d98c7
core/conversion/conversion.cpp
@@ -487,7 +487,8 @@ std::string ConvertBlockToEngine(
487
std::unordered_map<c10::OperatorName, std::string> GetUnsupportedOpsInBlock(const torch::jit::Block* b) {
488
std::unordered_map<c10::OperatorName, std::string> unsupported_ops;
489
for (const auto n : b->nodes()) {
490
- 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) &&
491
+ n->kind() != torch::jit::prim::DictConstruct) {
492
auto schema = n->maybeSchema();
493
TORCHTRT_CHECK(
494
schema,
0 commit comments