FM / utils

/usr/local/lib64/python3.6/site-packages/torch/utils/benchmark/utils/

/usr/local/lib64/python3.6/site-packages/torch/utils/benchmark/utils

Name	Size	Mode	Actions
valgrind_wrapper/	-	0755	rm
__pycache__/	-	0755	rm
common.py	13596	0644	edit dl rm
compare.py	12354	0644	edit dl rm
cpp_jit.py	6597	0644	edit dl rm
fuzzer.py	18211	0644	edit dl rm
sparse_fuzzer.py	5165	0644	edit dl rm
timeit_template.cpp	1017	0644	edit dl rm
timer.py	19292	0644	edit dl rm
_stubs.py	1142	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/utils/benchmark/utils/_stubs.py (1142B)

import sys
from typing import Any, Callable, Dict, TYPE_CHECKING

if TYPE_CHECKING or sys.version_info >= (3, 8):
    from typing import runtime_checkable, Protocol
else:
    from typing_extensions import runtime_checkable, Protocol

class TimerClass(Protocol):
    """This is the portion of the `timeit.Timer` API used by benchmark utils."""
    def __init__(
        self,
        stmt: str,
        setup: str,
        timer: Callable[[], float],
        globals: Dict[str, Any],
        **kwargs: Any,
    ) -> None:
        ...

def timeit(self, number: int) -> float:
        ...

@runtime_checkable
class TimeitModuleType(Protocol):
    """Modules generated from `timeit_template.cpp`."""
    def timeit(self, number: int) -> float:
        ...

class CallgrindModuleType(Protocol):
    """Replicates the valgrind endpoints in `torch._C`.

These bindings are used to collect Callgrind profiles on earlier versions
    of PyTorch and will eventually be removed.
    """
    __file__: str
    __name__: str

def _valgrind_supported_platform(self) -> bool:
        ...

def _valgrind_toggle(self) -> None:
        ...

import sys from typing import Any, Callable, Dict, TYPE_CHECKING if TYPE_CHECKING or sys.version_info >= (3, 8): from typing import runtime_checkable, Protocol else: from typing_extensions import runtime_checkable, Protocol class TimerClass(Protocol): """This is the portion of the `timeit.Timer` API used by benchmark utils.""" def __init__( self, stmt: str, setup: str, timer: Callable[[], float], globals: Dict[str, Any], **kwargs: Any, ) -> None: ... def timeit(self, number: int) -> float: ... @runtime_checkable class TimeitModuleType(Protocol): """Modules generated from `timeit_template.cpp`.""" def timeit(self, number: int) -> float: ... class CallgrindModuleType(Protocol): """Replicates the valgrind endpoints in `torch._C`. These bindings are used to collect Callgrind profiles on earlier versions of PyTorch and will eventually be removed. """ __file__: str __name__: str def _valgrind_supported_platform(self) -> bool: ... def _valgrind_toggle(self) -> None: ...