mindspore.runtime.dispatch_threads_num

查看源文件
mindspore.runtime.dispatch_threads_num(threads_num)[源代码]

设置运行时要使用的线程数。

框架默认运行时的线程数为5。

参数:
  • threads_num (int) - 要设置的线程数。

样例:

>>> import mindspore as ms
>>> ms.set_device("Ascend", 1)
>>> ms.runtime.dispatch_threads_num(6)