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

GitOrigin-RevId: e8dc7c4229faad25760d313a90ae8d87d13998d2

fix(imperative): fix broadcast extra assert of output shape
GitOrigin-RevId: e8dc7c4229faad25760d313a90ae8d87d13998d2
5c84305f · Megvii Engine Team · 7790040e · 5c84305f
隐藏空白更改
内联并排

Showing with 0 addition and 1 deletion

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

未找到文件。
--- a/imperative/src/impl/ops/broadcast.cpp
+++ b/imperative/src/impl/ops/broadcast.cpp
@@ -290,7 +290,6 @@ std::tuple<SmallVector<LogicalTensorDesc>, bool> infer_output_attrs_fallible(
        }
    }
    if (op.axis != opr::Reshape::Param::INVALID_AXIS) {
-        mgb_assert(out_shape[op.axis] == -1);
        out_shape[op.axis] = 1;
        mgb_assert(
                src.layout.total_nr_elems() % out_shape.total_nr_elems() == 0,