流

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

XPU 流的包裝器。

XPU 流是屬於特定裝置的線性執行序列，獨立於其他流。它支援 with 語句作為上下文管理器，以確保 with 塊內的運算子在相應的流上執行。

引數

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

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

query()

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

返回值

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

返回型別

bool

record_event(event=None)

記錄一個事件。

引數

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

返回值

記錄的事件。

synchronize()

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

wait_event(event)

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

引數

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

wait_stream(stream)

與另一個流同步。

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

引數

stream (Stream) – 要同步的流。