Browse Source

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

GitOrigin-RevId: b294eead0a
tags/v1.3.0
Megvii Engine Team 4 years ago
parent
commit
bc3561c1ee
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      imperative/src/impl/ops/elemwise.cpp

+ 2
- 1
imperative/src/impl/ops/elemwise.cpp View File

@@ -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)


Loading…
Cancel
Save