From 8ed2077bcb612fa80ddf7843c727a1bd3b82bad5 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Tue, 12 Jan 2021 16:46:18 +0800
Subject: [PATCH] fix(imperative/interpreter): reset m_waitee on exception

GitOrigin-RevId: ecf5245d9c0849b326efc0a343b22017c13bf7dc
---
 imperative/src/impl/interpreter_impl.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/imperative/src/impl/interpreter_impl.cpp b/imperative/src/impl/interpreter_impl.cpp
index 3f242c1bd..54ebd96e9 100644
--- a/imperative/src/impl/interpreter_impl.cpp
+++ b/imperative/src/impl/interpreter_impl.cpp
@@ -483,6 +483,8 @@ void ChannelImpl::process_one_task(Command& cmd) {
 
 void ChannelImpl::check_worker_exc_unsafe() {
     if (m_worker_exc) {
+        // for reuse interpreter_for_py after some exception tests
+        m_waitee = nullptr;
         std::exception_ptr exc;
         std::swap(exc, m_worker_exc);
         std::rethrow_exception(exc);
-- 
GitLab