mindspore.device_context.cpu.op_tuning.threads_numļƒ

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)