mindspore.runtime.StreamLimitCtx

class mindspore.runtime.StreamLimitCtx(stream, cube_num=- 1, vector_num=- 1)[源代码]

上下文管理器，用于选择给定流的核数限制。

上下文范围内的所有在给定流上执行的算子，都将指定cube和vector的核数。

参数：

stream (mindspore.runtime.Stream) - 指定的流。
cube_num (int，可选) - 设置流上的cube核数。默认值： -1，表示不设置。
vector_num (int，可选) - 设置流上的vector核数。默认值： -1，表示不设置。

异常：

TypeError - 参数 stream 不是 mindspore.runtime.Stream 。

支持平台：

Ascend

样例：

>>> import mindspore as ms
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> ms.set_device("Ascend", 0)
>>> a = Tensor(np.ones([1024, 2048]), ms.float32)
>>> b = Tensor(np.ones([2048, 4096]), ms.float32)
>>> s1 = ms.runtime.Stream()
>>> with ms.runtime.StreamLimitCtx(s1, 8, 8):
...     c = ops.matmul(a, b)
>>> ms.runtime.synchronize()