From 3b7c4cdff78c25dbfcf2c90c4dd7d0465b022400 Mon Sep 17 00:00:00 2001 From: xiefangqi Date: Wed, 14 Apr 2021 11:02:34 +0800 Subject: [PATCH] add numa interface to api docs --- mindspore/dataset/core/config.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/mindspore/dataset/core/config.py b/mindspore/dataset/core/config.py index 24436c8ef4..7492ddc717 100644 --- a/mindspore/dataset/core/config.py +++ b/mindspore/dataset/core/config.py @@ -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.