Skip to content

Commit 6391ae3

Browse files
committed
fix: [collection] skip ListConstruct in condition
Signed-off-by: inocsin <[email protected]>
1 parent 47455d1 commit 6391ae3

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

core/conversion/conversion.cpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -498,7 +498,9 @@ std::string ConvertBlockToEngine(
498498
std::unordered_map<c10::OperatorName, std::string> GetUnsupportedOpsInBlock(const torch::jit::Block* b) {
499499
std::unordered_map<c10::OperatorName, std::string> unsupported_ops;
500500
for (const auto n : b->nodes()) {
501-
if (n->kind() != torch::jit::prim::Loop && n->kind() != torch::jit::prim::If && !OpSupported(n) && n->kind() != torch::jit::prim::TupleConstruct) {
501+
if (n->kind() != torch::jit::prim::Loop && n->kind() != torch::jit::prim::If &&
502+
!OpSupported(n) && n->kind() != torch::jit::prim::TupleConstruct &&
503+
n->kind() != torch::jit::prim::ListConstruct) {
502504
auto schema = n->maybeSchema();
503505
TORCHTRT_CHECK(
504506
schema,

0 commit comments

Comments
 (0)