File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -498,7 +498,9 @@ std::string ConvertBlockToEngine(
498
498
std::unordered_map<c10::OperatorName, std::string> GetUnsupportedOpsInBlock (const torch::jit::Block* b) {
499
499
std::unordered_map<c10::OperatorName, std::string> unsupported_ops;
500
500
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) {
502
504
auto schema = n->maybeSchema ();
503
505
TORCHTRT_CHECK (
504
506
schema,
You can’t perform that action at this time.
0 commit comments