Shortcuts

事件

class torch.xpu.Event(enable_timing=False)[源代码]

围绕XPU事件的包装器。

XPU事件是同步标记,可用于监控设备的进度,并同步XPU流。

底层XPU事件在首次记录事件时被延迟初始化。创建后,只有同一设备上的流可以记录该事件。然而,任何设备上的流都可以等待该事件。

Parameters

enable_timing (bool, 可选) – 指示事件是否应测量时间 (默认值: False)

elapsed_time(end_event)[源代码]

返回已用时间。

事件记录后和end_event记录前的毫秒数报告。

query()[源码]

检查事件捕获的所有工作是否已完成。

Returns

一个布尔值,指示事件当前捕获的所有工作是否已完成。

record(stream=None)[源代码]

在给定的流中记录事件。

如果未指定流,则使用 torch.xpu.current_stream()。流的设备必须与事件的设备匹配。

synchronize()[源代码]

等待事件完成。

等待当前事件中捕获的所有工作完成。 这会阻止CPU线程继续执行,直到事件完成。

wait(stream=None)[源代码]

使所有提交到给定流的未来工作等待此事件。

如果没有指定流,请使用 torch.xpu.current_stream()

优云智算