From 5c84305f199a5d8314886b5081e13c5f13ef963d Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Sun, 9 Oct 2022 16:01:11 +0800
Subject: [PATCH] fix(imperative): fix broadcast extra assert of output shape

GitOrigin-RevId: e8dc7c4229faad25760d313a90ae8d87d13998d2
---
 imperative/src/impl/ops/broadcast.cpp | 1 -
 1 file changed, 1 deletion(-)

diff --git a/imperative/src/impl/ops/broadcast.cpp b/imperative/src/impl/ops/broadcast.cpp
index ff35cfed9..d2cb2c4b3 100644
--- 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,
-- 
GitLab