流

class torch.xpu.Stream(device=None, priority=0, **kwargs)

XPU流的包裝器。

XPU流是屬於特定裝置的一系列獨立執行操作。它支援with語句作為上下文管理器，以確保with塊內的操作在相應的流上執行。

引數

• device (torch.device 或 int, 可選) – 用於分配流的裝置。如果device為None（預設）或負整數，則將使用當前裝置。

• priority (int, 可選) – 流的優先順序，可以是正數、0或負數。數字越小表示優先順序越高。預設情況下，優先順序設定為0。如果值超出允許的優先順序範圍，它將自動對映到最近的有效優先順序（對於大的正數是最低優先順序，對於大的負數是最高優先順序）。

query()

檢查所有提交的工作是否已完成。

返回

一個布林值，指示此流中的所有核心是否已完成。

返回型別

布林值

record_event(event=None)

Record an event.

引數

event (torch.xpu.Event, 可選) – 要記錄的事件。如果未提供，將分配一個新的事件。

返回

記錄的事件。

synchronize()

等待此流中的所有核心完成。

wait_event(event)

使提交到此流的所有未來工作等待一個事件。

引數

event (torch.xpu.Event) – 要等待的事件。

wait_stream(stream)

與另一個流同步。

提交到此流的所有未來工作將等待，直到呼叫時提交到給定流的所有核心完成。

引數

stream (Stream) – 要同步的流。