From 7de1bb11ab357f328a1c9f851e10e2161f82bd7e Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Fri, 18 Mar 2022 16:20:48 +0800
Subject: [PATCH] fix(mge/utils): disable memory forwarding for subgraph

GitOrigin-RevId: 3be6d5fe6bfa933eb6d4b4dd38acdb2629b31755
---
 imperative/src/impl/ops/utility.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/imperative/src/impl/ops/utility.cpp b/imperative/src/impl/ops/utility.cpp
index d39a5958..37d4cfaf 100644
--- a/imperative/src/impl/ops/utility.cpp
+++ b/imperative/src/impl/ops/utility.cpp
@@ -411,7 +411,7 @@ struct ComputingGraphHolder {
         executable->wait();
         size_t nr_inputs = inputs.size();
         for (size_t i = 0; i < nr_inputs; ++i) {
-            auto input_dev_tensor = input_tensors[i]->dev_tensor(false);
+            auto input_dev_tensor = input_tensors[i]->dev_tensor(true);
             inputs[i].device_value->reset(
                     input_dev_tensor.storage(), input_dev_tensor.layout());
             if (inputs[i].host_value) {