xiefangqi
5 years ago
1 changed files with 4 additions and 3 deletions
Unified View
Diff Options
+ 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 | from mindspore import log as logger | ||||
| __all__ = ['set_seed', 'get_seed', 'set_prefetch_size', 'get_prefetch_size', 'set_num_parallel_workers', | __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 | INT32_MAX = 2147483647 | ||||
| UINT32_MAX = 4294967295 | UINT32_MAX = 4294967295 | ||||
| @@ -176,7 +177,7 @@ def get_num_parallel_workers(): | |||||
| def set_numa_enable(numa_enable): | 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: | Args: | ||||
| numa_enable (bool): Whether to use numa bind feature. | numa_enable (bool): Whether to use numa bind feature. | ||||