围绕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()。