mindspore.hal.StreamCtx

class mindspore.hal.StreamCtx(ctx_stream)[源代码]

上下文管理器，用于选择给定的流，此接口将在后续版本中废弃，请使用接口 mindspore.runtime.StreamCtx 代替。

在上下文范围内，所有算子都将在指定流上执行。

参数：

ctx_stream (Stream) - 指定的流。如果是 None ，则该上下文管理器无操作。

样例：

>>> import mindspore
>>> a = mindspore.tensor(mindspore.ops.ones([1024, 2048]), mindspore.float32)
>>> b = mindspore.tensor(mindspore.ops.ones([2048, 4096]), mindspore.float32)
>>> s1 = mindspore.hal.Stream()
>>> with mindspore.hal.StreamCtx(s1):
...     c = mindspore.ops.matmul(a, b)
>>> mindspore.hal.synchronize()
>>> assert s1.query()