|
|
|
@@ -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. |
|
|
|
|