
    '}h                     t    d dl Z d dlZd dlZddlmZmZ g dZg dZd
dZd Z	d Z
e j                  d	        Zy)    N   )check_errorcudart)initstartstopprofile)gpustarttimestampgpuendtimestamp
gridsize3dthreadblocksizestreamidzenableonstart 0conckerneltracec                    t               }t        |d      st        d      t        t        j                  d      r]t        j                  j
                  Ct        t        j                  j
                  j                  d      d         dk\  rt        d      |t        n|}|dk(  r|j                  j                  }n'|d	k(  r|j                  j                  }nt        d
      t        j                  d      5 }|j                  dj!                  d |D                     |j#                          t%        |j'                  |j(                  | |             d d d        y # 1 sw Y   y xY w)NcudaOutputModez-HIP does not support profiler initialization!cuda.r      z.CUDA12+ does not need profiler initialization!	key_valuecsvz;supported CUDA profiler output modes are: key_value and csvT)delete   
c              3   >   K   | ]  }|j                  d         yw)asciiN)encode).0fs     R/var/www/html/test/engine/venv/lib/python3.12/site-packages/torch/cuda/profiler.py	<genexpr>zinit.<locals>.<genexpr>*   s     <188G,<s   )r   hasattrAssertionErrortorchversionr   intsplitDEFAULT_FLAGSr   KeyValuePairCSVRuntimeErrortempfileNamedTemporaryFilewritejoinflushr   cudaProfilerInitializename)output_fileflagsoutput_modertoutput_mode_enumr   s         r   r   r      s-   	B2'(LMMv&MM*""((-a01R7 MNN"]MEk!,,99		,,00I
 	
 
	$	$D	1 VQ	

<e<<=		B--affkCSTUV V Vs   <AE##E,c                  F    t        t               j                                y N)r   r   cudaProfilerStart     r   r   r   /   s    **,-r:   c                  F    t        t               j                                y r7   )r   r   cudaProfilerStopr9   r:   r   r   r   3   s    ))+,r:   c               #   \   K   	 t                d  t                y # t                w xY wwr7   )r   r   r9   r:   r   r	   r	   7   s     s   , ,),)Nr   )
contextlibr*   r"    r   r   __all__r&   r   r   r   contextmanagerr	   r9   r:   r   <module>rB      sG       !
.V6.-  r:   