提交 bc3561c1 编写于 作者: M Megvii Engine Team

fix(imperative/ops): fix inplace_add's output desc, remove value

GitOrigin-RevId: b294eead0a1c12742e825182cf96e60acf8bf68c
上级 6295f843
......@@ -208,7 +208,8 @@ std::tuple<SmallVector<LogicalTensorDesc>, bool> infer_inplace_add_output_attrs_
}
mgb_assert(alpha.layout.dtype == dtype::Float32(), "alpha should be float32");
mgb_assert(beta.layout.dtype == dtype::Float32(), "beta should be float32");
return {{dest}, succeed};
// inplace op result's desc value is changed
return {{{dest.layout, dest.comp_node}}, succeed};
}
OP_TRAIT_REG(Elemwise, Elemwise, opr::Elemwise)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册