From bff0fc6172a3c537cc8d992e2bae0477813ec785 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Mon, 7 Dec 2020 15:04:09 +0800
Subject: [PATCH] fix(mge/interpreter): fix outputs check on async level0

GitOrigin-RevId: 46c6a2ce7062d2ee0920c2e1ea9047536f475ac5
---
 imperative/src/impl/interpreter_impl.cpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/imperative/src/impl/interpreter_impl.cpp b/imperative/src/impl/interpreter_impl.cpp
index 70be59855..650b846b4 100644
--- a/imperative/src/impl/interpreter_impl.cpp
+++ b/imperative/src/impl/interpreter_impl.cpp
@@ -132,8 +132,9 @@ SmallVector<void*> ChannelImpl::apply_op(
     } else if (m_async_level == 0) {
         sync();
         // check device error
-        for (auto&& oup : cmd.outputs) {
-            oup->ptr->comp_node().sync();
+        for (auto&& oup : outputs) {
+            auto info = reinterpret_cast<TensorInfo*>(oup);
+            info->ptr->comp_node().sync();
         }
     }
     return outputs;
-- 
GitLab