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

fix(mgb/opr): add layout constraint on the input of Cumsum

GitOrigin-RevId: d2d108e2786806aa3f4c44ebfc0bf5f094cdab42
上级 0d165399
......@@ -132,6 +132,10 @@ void Cumsum::scn_do_execute() {
intl::get_megdnn_workspace_from_var(output().back()));
}
void Cumsum::add_input_layout_constraint() {
input(0)->add_layout_constraint_contiguous();
}
void Cumsum::init_output_static_infer_desc() {
using namespace cg::static_infer;
auto infer_shape = [](TensorShape& dest, const InpVal& iv) {
......
......@@ -86,6 +86,7 @@ public:
//! cumulative sum along given axis
MGB_DEFINE_OPR_CLASS(Cumsum, cg::SingleCNOperatorNodeBaseT<
mixin::MegDNNOprHolderImpl<megdnn::Cumsum>>) // {
void add_input_layout_constraint() override;
public:
Cumsum(VarNode *src, const Param &param, const OperatorNodeConfig &config);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册