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 0b8321b commit 5c26000Copy full SHA for 5c26000
core/conversion/conversion.cpp
@@ -487,8 +487,7 @@ 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) &&
491
- n->kind() != torch::jit::prim::DictConstruct) {
+ if (n->kind() != torch::jit::prim::Loop && n->kind() != torch::jit::prim::If && !OpSupported(n)) {
492
auto schema = n->maybeSchema();
493
TORCHTRT_CHECK(
494
schema,
0 commit comments