Name	Size	Mode	Actions
comm.cpython-36.pyc	372	0644	editdlrm
error.cpython-36.pyc	143	0644	editdlrm
graphs.cpython-36.pyc	15811	0644	editdlrm
memory.cpython-36.pyc	21717	0644	editdlrm
nccl.cpython-36.pyc	3444	0644	editdlrm
nvtx.cpython-36.pyc	1966	0644	editdlrm
profiler.cpython-36.pyc	1632	0644	editdlrm
random.cpython-36.pyc	5520	0644	editdlrm
sparse.cpython-36.pyc	144	0644	editdlrm
streams.cpython-36.pyc	9485	0644	editdlrm
_utils.cpython-36.pyc	1892	0644	editdlrm
__init__.cpython-36.pyc	23210	0644	editdlrm

3 Eg��@s^dd
lZdd
l
Z
ddlmZmZ
dddddd d gZddd �
Zdd�Zdd�Ze
j dd��
Z d
S)�N�
)�cudart�check_errorZgpustarttimestampZgpuendtimestampZ gridsize3dZthreadblocksizeZstreamidzenableonstart 0Zconckerneltrace� key_valuecCs�t�}t
|d
�std�
�
|
dkr$tn|
}
|dkr:|jj}n|dkrL|jj}ntd�
�
tj dd�
�<}|j djd d �|
D�
�
�

|j�
t |j|j||��

WdQRXdS)N�cudaOutputModez-HIP does not support profiler initialization!r�csvz;supported CUDA profiler output modes are: key_value and csvT)
�delete�
c
ss|]}
|
jd�
V
qd
S)�asciiN)
�encode)�.0�
f�r�?/usr/local/lib64/python3.6/site-packages/torch/cuda/profiler.py� szinit..)r�hasattr�AssertionError� DEFAULT_FLAGSrZKeyValuePairZCSV�RuntimeError�tempfile�NamedTemporaryFile�write�join�flushrZcudaProfilerInitialize�name)Zoutput_file�flagsZoutput_mode�rtZoutput_mode_enumr rrr�inits











rcCstt
�j��

dS)
N)rrZcudaProfilerStartrrrr�start"s
rcCstt
�j��

dS)
N)rrZcudaProfilerStoprrrr�stop&s
rc cszt�
dV
Wdt
�
XdS)
N)rrrrrr�profile*s

r )Nr)r� contextlib�rrrrrr�contextmanagerr rrrr�
s