StreamContext

class torch.mtia.StreamContext(stream) [源代码]

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

所有在上下文中排队的 MTIA 内核将被加入选定的流中。

参数

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

注意

流是针对每个设备的。