Browse Source

!15143 set_numa_enable and get_numa_enable are not in the API doc

From: @xiefangqi
Reviewed-by: @liucunwei,@heleiwang
Signed-off-by: @liucunwei
pull/15143/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
3ceeeceba1
1 changed files with 4 additions and 3 deletions
  1. +4
    -3
      mindspore/dataset/core/config.py

+ 4
- 3
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.


Loading…
Cancel
Save