mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
dcfa7702c3
See #127836 for details. Pull Request resolved: https://github.com/pytorch/pytorch/pull/127838 Approved by: https://github.com/oulgen
8 lines
250 B
Python
8 lines
250 B
Python
# mypy: allow-untyped-defs
|
|
# Defined in torch/csrc/cuda/shared/nvtx.cpp
|
|
def rangePushA(message: str) -> int: ...
|
|
def rangePop() -> int: ...
|
|
def rangeStartA(message: str) -> int: ...
|
|
def rangeEnd(int) -> None: ...
|
|
def markA(message: str) -> None: ...
|