!10314 [MD] fix pool not running bug

From: @liyong126
Reviewed-by: @liucunwei,@liucunwei
Signed-off-by: @liucunwei,@liucunwei
This commit is contained in:
mindspore-ci-bot 2020-12-23 16:24:33 +08:00 committed by Gitee
commit 26a7eb9e49
1 changed files with 1 additions and 1 deletions

View File

@ -2211,7 +2211,7 @@ class _PythonCallable:
self.idx = idx
def __call__(self, *args):
if self.pool is not None:
if self.pool is not None and self.pool._state == 0: # pylint: disable=W0212
# This call will send the tensors along with Python callable index to the process pool.
# Block, yield GIL. Current thread will reacquire GIL once result is returned.
result = self.pool.apply_async(_pyfunc_worker_exec, [self.idx, *args])