mindspore.runtime.get_stream_limit
- mindspore.runtime.get_stream_limit(stream)[源代码]
返回指定流上限制核数。
说明
该接口会同步下发和执行,可能会影响性能。
当前仅支持PyNative模式,不支持Graph模式。
- 参数:
stream (
mindspore.runtime.Stream) - 指定的流对象。
- 返回:
dict,查询到流上的限制核数信息。
- 支持平台:
Ascend
样例:
>>> import mindspore as ms >>> ms.runtime.get_stream_limit(ms.runtime.default_stream())