mindspore.runtime.dispatch_threads_num

View Source On Gitee
mindspore.runtime.dispatch_threads_num(threads_num)[source]

Set the threads number of runtime used.

The framework set the runtime number of threads are 5 by default.

Parameters

threads_num (int) – The threads number of runtime used.

Examples

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