OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
Browse Source

fix(imperative): fix broadcast extra assert of output shape 

GitOrigin-RevId: e8dc7c4229
master
Megvii Engine Team 2 years ago
parent
7790040e23
commit
5c84305f19
1 changed files with 0 additions and 1 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +0
    -1
      imperative/src/impl/ops/broadcast.cpp

+ 0
- 1
imperative/src/impl/ops/broadcast.cpp View File

@@ -290,7 +290,6 @@ std::tuple<SmallVector<LogicalTensorDesc>, bool> infer_output_attrs_fallible(
} }
} }
if (op.axis != opr::Reshape::Param::INVALID_AXIS) { if (op.axis != opr::Reshape::Param::INVALID_AXIS) {
mgb_assert(out_shape[op.axis] == -1);
out_shape[op.axis] = 1; out_shape[op.axis] = 1;
mgb_assert( mgb_assert(
src.layout.total_nr_elems() % out_shape.total_nr_elems() == 0, src.layout.total_nr_elems() % out_shape.total_nr_elems() == 0,


Write Preview
Loading…
Cancel
Save