pytorch

1
from abc import ABC, abstractmethod
2

3

4
class _StreamBase(ABC):
5
    r"""Base stream class abstraction for multi backends Stream to herit from"""
6

7
    @abstractmethod
8
    def wait_event(self, event):
9
        raise NotImplementedError()
10

11
    @abstractmethod
12
    def wait_stream(self, stream):
13
        raise NotImplementedError()
14

15
    @abstractmethod
16
    def record_event(self, event=None):
17
        raise NotImplementedError()
18

19
    @abstractmethod
20
    def query(self):
21
        raise NotImplementedError()
22

23
    @abstractmethod
24
    def synchronize(self):
25
        raise NotImplementedError()
26

27
    @abstractmethod
28
    def __eq__(self, stream):
29
        raise NotImplementedError()
30

31

32
class _EventBase(ABC):
33
    r"""Base Event class abstraction for multi backends Event to herit from"""
34

35
    @abstractmethod
36
    def wait(self, stream=None):
37
        raise NotImplementedError()
38

39
    @abstractmethod
40
    def query(self):
41
        raise NotImplementedError()
42

43
    @abstractmethod
44
    def synchronize(self):
45
        raise NotImplementedError()
46