mindspore.runtime.get_stream_limit

View Source On Gitee
mindspore.runtime.get_stream_limit(stream)[source]

Return selected stream limit core num.

Note

  • This interface will synchronize the operator issuance, which may affect performance.

  • Only support PyNative mode, Graph mode is not currently supported.

Parameters

stream (mindspore.runtime.Stream) – selected stream.

Returns

limit info (dict), stream limit core num.

Supported Platforms:

Ascend

Examples

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