/usr/local/lib64/python3.6/site-packages/torch/cuda/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/torch/cuda/__pycache__/nvtx.cpython-36.pyc (1966B)
3
Egä ã @ s| d dl mZ yd dlmZ W n* ek
rF G dd„ deƒZeƒ ZY nX ddddgZd d„ Zd
d„ Z d d„ Z
ed d„ ƒZ d
S )é )Úcontextmanager)Ú_nvtxc @ s$ e Zd Zedd„ ƒZeZeZeZdS )Ú _NVTXStubc O s t dƒ‚d S )NzANVTX functions not installed. Are you sure you have a CUDA build?)Ú RuntimeError)ÚargsÚkwargs© r ú;/usr/local/lib64/python3.6/site-packages/torch/cuda/nvtx.pyÚ_fail s z_NVTXStub._failN)Ú__name__Ú
__module__Ú __qualname__Ú staticmethodr
Ú
rangePushAÚrangePopÚmarkAr r r r r s r Ú
range_pushÚ range_popÚmarkÚrangec C s
t j| ƒS )z¾
Pushes a range onto a stack of nested range span. Returns zero-based
depth of the range that is started.
Args:
msg (string): ASCII message to associate with range
)r r )Úmsgr r r r s c C s t jƒ S )zz
Pops a range off of a stack of nested range spans. Returns the
zero-based depth of the range that is ended.
)r r r r r r r s c C s
t j| ƒS )z’
Describe an instantaneous event that occurred at some point.
Args:
msg (string): ASCII message to associate with the event.
)r r )r r r r r ' s c o s t | j||Žƒ dV tƒ dS )a
Context manager / decorator that pushes an NVTX range at the beginning
of its scope, and pops it at the end. If extra arguments are given,
they are passed as arguments to msg.format().
Args:
msg (string): message to associate with the range
N)r Úformatr )r r r r r r r 1 s
N) Ú
contextlibr Ztorch._Cr Ú ImportErrorÚobjectr Ú__all__r r r r r r r r Ú
s