StreamContext

class torch.xpu.StreamContext(stream)

選擇給定 stream 的上下文管理器。

在其上下文中排隊的所有 XPU 核心都將在選定的 stream 上排隊。

參數

Stream (Stream) – 選定的 stream。如果它是 None，則此管理器不執行任何操作。

注意

Stream 是每個裝置。