StreamContext

class torch.mtia.StreamContext(stream)

選擇給定串流的環境管理器。

在其環境中排隊的所有 MTIA 核心都將排隊在選定的串流上。

參數

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

注意

串流是每個裝置的。