mindspore.device_context.cpu.op_tuning.threads_num.rst

查看源文件
mindspore.device_context.cpu.op_tuning.threads_num(num)[源代码]

设置CPU算子使用的线程数。

框架默认设置的线程数为 25

参数:
  • num (int) - CPU算子使用的线程数。

样例:

>>> import mindspore as ms
>>> ms.set_device("Ascend", 1)
>>> ms.device_context.cpu.op_tuning.threads_num(10)