mindspore.device_context.cpu.op_tuning.threads_num

View Source On AtomGit
mindspore.device_context.cpu.op_tuning.threads_num(num)[source]

Set the number of threads for CPU kernel used.

The number of CPU kernel threads used by the framework is 25 by default.

Parameters

num (int) – Number of threads for CPU kernel.

Examples

>>> import mindspore as ms
>>> ms.set_device("CPU")
>>> ms.device_context.cpu.op_tuning.threads_num(10)