a NSic ã@s’ddlmZzddlmZWn(eyDGdd„deƒZeƒZYn0gd¢Zdd„Zdd „Z e d œd d „Z d d œdd„Z dd„Z edd„ƒZd S)é)Úcontextmanager)Ú_nvtxc@s$eZdZedd„ƒZeZeZeZdS)Ú _NVTXStubcOs tdƒ‚dS)NzANVTX functions not installed. Are you sure you have a CUDA build?)Ú RuntimeError)ÚargsÚkwargs©rúK/var/www/html/django/DPS/env/lib/python3.9/site-packages/torch/cuda/nvtx.pyÚ_failsz_NVTXStub._failN)Ú__name__Ú __module__Ú __qualname__Ú staticmethodr Ú rangePushAÚrangePopÚmarkArrrr rs  r)Ú range_pushÚ range_popÚ range_startÚ range_endÚmarkÚrangecCs t |¡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 )rr©Úmsgrrr rsrcCst ¡S)zz Pops a range off of a stack of nested range spans. Returns the zero-based depth of the range that is ended. )rrrrrr rsr)ÚreturncCs t |¡S)aì Mark the start of a range with string message. It returns an unique handle for this range to pass to the corresponding call to rangeEnd(). A key difference between this and range_push/range_pop is that the range_start/range_end version supports range across threads (start on one thread and end on another thread). Returns: A range handle (uint64_t) that can be passed to range_end(). Args: msg (string): ASCII message to associate with the range. )rÚ rangeStartArrrr r'srNcCst |¡dS)z€ Mark the end of a range for a given range_id. Args: range_id (int): an unique handle for the start range. N)rÚrangeEnd)Zrange_idrrr r8srcCs t |¡S)z’ Describe an instantaneous event that occurred at some point. Args: msg (string): ASCII message to associate with the event. )rrrrrr rBsrcos$t|j|i|¤ŽƒdVtƒ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)rrrrrr rLs r)Ú contextlibrÚtorch._CrÚ ImportErrorÚobjectrÚ__all__rrÚintrrrrrrrr Ús