|
|
|
@@ -102,9 +102,20 @@ class TbeProcess: |
|
|
|
def __init__(self): |
|
|
|
self.__processe_num = multiprocessing.cpu_count() |
|
|
|
# max_processes_num: Set the maximum number of concurrent processes for compiler |
|
|
|
max_processes_num = 24 |
|
|
|
if self.__processe_num > max_processes_num: |
|
|
|
self.__processe_num = max_processes_num |
|
|
|
self.max_processes_num = 24 |
|
|
|
process_num = os.getenv("MS_BUILD_PROCESS_NUM") |
|
|
|
if process_num is None: |
|
|
|
self.max_processes_num = 24 |
|
|
|
elif process_num.isdigit(): |
|
|
|
if int(process_num) in range(1, 25): |
|
|
|
self.max_processes_num = int(process_num) |
|
|
|
else: |
|
|
|
raise EnvironmentError( |
|
|
|
f"Env ERROR, [MS_BUILD_PROCESS_NUM] should be in range(1, 25), but: {process_num}") |
|
|
|
elif not process_num.isdigit(): |
|
|
|
raise EnvironmentError(f"Env ERROR, [MS_BUILD_PROCESS_NUM] should be a digit, but: {process_num}") |
|
|
|
if self.__processe_num > self.max_processes_num: |
|
|
|
self.__processe_num = self.max_processes_num |
|
|
|
self.__pool = None |
|
|
|
self.__next_task_id = 1 |
|
|
|
self.__running_tasks = [] |
|
|
|
|