mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!15143 set_numa_enable and get_numa_enable are not in the API doc 

From: @xiefangqi
Reviewed-by: @liucunwei,@heleiwang
Signed-off-by: @liucunwei
This commit is contained in:
mindspore-ci-bot 2021-04-14 15:53:41 +08:00 committed by Gitee
parent 9ef83fe7f4 3b7c4cdff7
commit 3ceeeceba1
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mindspore/dataset/core/config.py
View File

@ -24,8 +24,9 @@ import mindspore._c_dataengine as cde
from mindspore import log as logger
__all__ = ['set_seed', 'get_seed', 'set_prefetch_size', 'get_prefetch_size', 'set_num_parallel_workers',
'get_num_parallel_workers', 'set_monitor_sampling_interval', 'get_monitor_sampling_interval', 'load',
'get_callback_timeout', 'set_auto_num_workers', 'get_auto_num_workers', '_init_device_info']
'get_num_parallel_workers', 'set_numa_enable', 'get_numa_enable', 'set_monitor_sampling_interval',
'get_monitor_sampling_interval', 'load', 'get_callback_timeout', 'set_auto_num_workers',
'get_auto_num_workers', '_init_device_info']
INT32_MAX = 2147483647
UINT32_MAX = 4294967295
@ -176,7 +177,7 @@ def get_num_parallel_workers():
def set_numa_enable(numa_enable):
"""
Set the default state of numa enabled.
Set the default state of numa enabled. If numa_enable is True, need to ensure numa library is installed.
Args:
numa_enable (bool): Whether to use numa bind feature.