提交 80dcd70e 编写于 作者: Z zhaoying 提交者: jackzhang235

(bugfix): args for some op(for example, conv2d) does not has type attr

上级 f07892b6
...@@ -104,7 +104,7 @@ void Int64ToInt32Pass::ChangeInt64ToInt32IfNeeded(Node* inst_node) { ...@@ -104,7 +104,7 @@ void Int64ToInt32Pass::ChangeInt64ToInt32IfNeeded(Node* inst_node) {
for (auto* in : inst_node->inlinks) { for (auto* in : inst_node->inlinks) {
CHECK(in->IsRoleSet()); CHECK(in->IsRoleSet());
CHECK(in->IsArg()); CHECK(in->IsArg());
CHECK(in->AsArg().type); if (in->AsArg().type) {
auto in_arg_name = in->AsArg().name; auto in_arg_name = in->AsArg().name;
std::string tmp; std::string tmp;
CHECK(inst.op_info()->GetInputArgname(in_arg_name, &tmp)); CHECK(inst.op_info()->GetInputArgname(in_arg_name, &tmp));
...@@ -124,11 +124,12 @@ void Int64ToInt32Pass::ChangeInt64ToInt32IfNeeded(Node* inst_node) { ...@@ -124,11 +124,12 @@ void Int64ToInt32Pass::ChangeInt64ToInt32IfNeeded(Node* inst_node) {
in->AsArg().type->layout())); in->AsArg().type->layout()));
} }
} }
}
for (auto* out : inst_node->outlinks) { for (auto* out : inst_node->outlinks) {
CHECK(out->IsRoleSet()); CHECK(out->IsRoleSet());
CHECK(out->IsArg()); CHECK(out->IsArg());
CHECK(out->AsArg().type); if (out->AsArg().type) {
auto out_arg_name = out->AsArg().name; auto out_arg_name = out->AsArg().name;
std::string tmp; std::string tmp;
CHECK(inst.op_info()->GetOutputArgname(out_arg_name, &tmp)); CHECK(inst.op_info()->GetOutputArgname(out_arg_name, &tmp));
...@@ -147,6 +148,7 @@ void Int64ToInt32Pass::ChangeInt64ToInt32IfNeeded(Node* inst_node) { ...@@ -147,6 +148,7 @@ void Int64ToInt32Pass::ChangeInt64ToInt32IfNeeded(Node* inst_node) {
out->AsArg().type->layout())); out->AsArg().type->layout()));
} }
} }
}
} }
} // namespace mir } // namespace mir
} // namespace lite } // namespace lite
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册