From bc3561c1eed5363fb2d07caabe4ac53c0bf9855e Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Tue, 12 Jan 2021 16:28:18 +0800
Subject: [PATCH] fix(imperative/ops): fix inplace_add's output desc, remove
 value

GitOrigin-RevId: b294eead0a1c12742e825182cf96e60acf8bf68c
---
 imperative/src/impl/ops/elemwise.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/imperative/src/impl/ops/elemwise.cpp b/imperative/src/impl/ops/elemwise.cpp
index b33b5cd1..e32c10d8 100644
--- a/imperative/src/impl/ops/elemwise.cpp
+++ b/imperative/src/impl/ops/elemwise.cpp
@@ -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)
-- 
GitLab