事件¶ class torch.xpu.Event(enable_timing=False)[source]¶ XPU 事件的包装器。 XPU 事件是同步标记,可用于监控设备的进度,以及同步 XPU 流。 当事件第一次记录时,底层 XPU 事件会被延迟初始化。创建后,只有同一设备上的流才能记录事件。但是,任何设备上的流都可以等待事件。 参数 enable_timing (布尔值, 可选) – 指示事件是否应测量时间(默认值:False) elapsed_time(end_event)[source]¶ 返回经过的时间。 记录事件后和记录 end_event 之前以毫秒为单位报告的时间。 query()[source]¶ 检查事件当前捕获的所有工作是否已完成。 返回值 一个布尔值,指示事件当前捕获的所有工作是否已完成。 返回类型 布尔值 record(stream=None)[source]¶ 在给定流中记录事件。 如果未指定流,则使用 torch.xpu.current_stream()。流的设备必须与事件的设备匹配。 synchronize()[source]¶ 等待事件完成。 等待此事件中当前捕获的所有工作完成。这会阻止 CPU 线程在事件完成之前继续执行。 wait(stream=None)[source]¶ 使提交到给定流的所有未来工作等待此事件。 如果未指定流,则使用 torch.xpu.current_stream()。