StreamContext

类torch.xpu.StreamContext(stream)[源代码]

用于选择指定流的上下文管理器。

其上下文中所有排队的XPU内核将会被添加到选定的流中。

参数

Stream (Stream) – 选定的流。如果为None，则该管理器不会执行任何操作。

注意

流是针对每个设备的。