mindspore.runtime.get_stream_limit

查看源文件
mindspore.runtime.get_stream_limit(stream)[源代码]

返回指定流上限制核数。

说明

  • 该接口会同步下发和执行,可能会影响性能。

  • 当前仅支持PyNative模式,不支持Graph模式。

参数:
返回:

dict,查询到流上的限制核数信息。

支持平台:

Ascend

样例:

>>> import mindspore as ms
>>> ms.runtime.get_stream_limit(ms.runtime.default_stream())