OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[CPU Stream] Add noop for CPU stream record_event() and wait_event() (#145935)

Browse Source 
Summary: Adds wait_event and record_event endpoints to CPU stream in order to facilitate device-agnostic code. Both methods are noops.

Test Plan: CI

Differential Revision: D68833927

Pull Request resolved: https://github.com/pytorch/pytorch/pull/145935
Approved by: https://github.com/Skylion007
This commit is contained in:
Jessica Vandebon 2025-02-20 18:50:55 +00:00 committed by PyTorch MergeBot
parent 863ac20659
commit 6971b77510
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
torch/cpu/__init__.py
View File

@ -95,6 +95,12 @@ class Stream:
def wait_stream(self, stream) -> None:
pass
def record_event(self) -> None:
pass
def wait_event(self, event) -> None:
pass
class Event:
def query(self) -> bool: