1# mypy: allow-untyped-defs
2# Defined in torch/csrc/cuda/shared/nvtx.cpp
3def rangePushA(message: str) -> int: ...
4def rangePop() -> int: ...
5def rangeStartA(message: str) -> int: ...
6def rangeEnd(int) -> None: ...
7def markA(message: str) -> None: ...
8