Stream

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

XPU stream 的包裝函式。

XPU stream 是一個屬於特定設備的線性執行序列，獨立於其他 stream。

參數

• device (torch.device 或 int, optional) – 用於分配 stream 的設備。如果 device 為 None (預設) 或負整數，這將使用目前的設備。

• priority (int, optional) – stream 的優先順序，應為 0 或負數，其中負數表示更高的優先順序。預設情況下，stream 的優先順序為 0。

query()

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

返回 (Returns)

一個布林值，表示此 stream 中的所有核心 (kernel) 是否已完成。

返回類型 (Return type)

bool

record_event(event=None)

記錄一個事件 (event)。

參數

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

返回 (Returns)

已記錄的事件 (event)。

synchronize()

等待此 stream 中的所有核心 (kernel) 完成。

wait_event(event)

使提交到 stream 的所有未來工作等待一個事件 (event)。

參數

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

wait_stream(stream)

與另一個 stream 同步。

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

參數

stream (Stream) – 要同步的 stream。