mindspore.dataset.config.set_enable_shared_mem

View Source On Gitee
mindspore.dataset.config.set_enable_shared_mem(enable)

Set whether to use shared memory for interprocess communication when data processing multiprocessing is turned on.

Using shared memory can speed up the efficiency of data transfer between processes.

Shared memory is used by default.

Note

  • Windows and MacOS systems are not supported yet.

  • The set_enable_shared_mem interface will be deprecated in a future version.

Parameters

enable (bool) – Whether to use shared memory for interprocess communication.

Raises

TypeError – If enable is not of type bool.

Examples

>>> # Enable shared memory feature to improve the performance of Python multiprocessing.
>>> import mindspore.dataset as ds
>>> ds.config.set_enable_shared_mem(True)