pytorch

1
# mypy: allow-untyped-defs
2
import types
3
from typing import NewType, Tuple
4

5
from torch._dynamo.types import DynamoCallback, DynamoGuardHook
6

7
# We implement our own FrameType-like type for Python >= 3.11. So it's not actually an alias of FrameType, but still
8
# exposes the same interface.
9
_PyInterpreterFrame = NewType("_PyInterpreterFrame", types.FrameType)
10

11
def set_eval_frame(callback: DynamoCallback) -> DynamoCallback: ...
12
def reset_code(code: types.CodeType) -> None: ...
13
def unsupported(obj1: object, obj2: object) -> object: ...
14
def skip_code(code: types.CodeType) -> None: ...
15
def set_guard_error_hook(hook: DynamoGuardHook) -> None: ...
16
def set_context_frame(context: Tuple[int, int, int]) -> None: ...
17

18
class _CacheEntry:
19
    def check_fn(self, *args, **kwargs): ...
20
    code: types.CodeType
21
    next: _CacheEntry | None
22

23
class _ExtraState:
24
    def invalidate(self, cache_entry: _CacheEntry): ...
25

26
def _debug_get_cache_entry_list(code: types.CodeType) -> list[_CacheEntry]: ...
27

28
py_opcode_caches: list[int]
29