From d3d9018f8da628e544644905ac05859e1a487986 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Tue, 14 Jul 2020 17:33:04 +0800
Subject: [PATCH] feat(imperative/opr-mm): add broadcast

GitOrigin-RevId: 83640255c758b85a4cbb7d40fb20b822d4a8b30f
---
 src/opr-mm/impl/collective_comm.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/opr-mm/impl/collective_comm.cpp b/src/opr-mm/impl/collective_comm.cpp
index 46e2b541..b8ea6fdf 100644
--- a/src/opr-mm/impl/collective_comm.cpp
+++ b/src/opr-mm/impl/collective_comm.cpp
@@ -810,7 +810,7 @@ void CollectiveComm::init_output_static_infer_desc() {
         };
 
         auto get_shape_from_server = [this](TensorShape& dest, const InpVal&) {
-            if (!m_enable_shape_infer) {
+            if (!m_enable_shape_infer && !owner_graph()->options().imperative_proxy_graph) {
                 return false;
             }
 
-- 
GitLab