Browse Source

!9 change parallel complie num: 32->16

Merge pull request !9 from jjfeing/master
tags/v0.2.0-alpha
mindspore-ci-bot Gitee 6 years ago
parent
commit
04eab166e5
1 changed files with 4 additions and 2 deletions
  1. +4
    -2
      mindspore/_extends/parallel_compile/tbe_compiler/tbe_process.py

+ 4
- 2
mindspore/_extends/parallel_compile/tbe_compiler/tbe_process.py View File

@@ -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 = []


Loading…
Cancel
Save