From 86f5c69995c3266d0570b53f3cedeb7ac1d2e5e3 Mon Sep 17 00:00:00 2001 From: jjfeing Date: Mon, 30 Mar 2020 10:03:48 +0800 Subject: [PATCH] change parallel complie num: 32->16 --- .../_extends/parallel_compile/tbe_compiler/tbe_process.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/mindspore/_extends/parallel_compile/tbe_compiler/tbe_process.py b/mindspore/_extends/parallel_compile/tbe_compiler/tbe_process.py index 91207b4f81..9b970b7242 100644 --- a/mindspore/_extends/parallel_compile/tbe_compiler/tbe_process.py +++ b/mindspore/_extends/parallel_compile/tbe_compiler/tbe_process.py @@ -150,8 +150,10 @@ class CompilerPool: def __init__(self): processes = multiprocessing.cpu_count() - if processes > 32: - processes = 32 + # max_processes_num: Set the maximum number of concurrent processes for compiler + max_processes_num = 16 + if processes > max_processes_num: + processes = max_processes_num self.__pool = multiprocessing.Pool(processes=processes) self.__next_task_id = 1 self.__running_tasks = []