StreamContext

class torch.mtia.StreamContext(stream)

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

在此上下文内排队的所有 MTIA kernel 都将排队到选定的流中。

参数

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

注意

流是针对每个设备的。