
    #h                     r    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dS )    N   )check_errorcudart)initstartstopprofile)gpustarttimestampgpuendtimestamp
gridsize3dthreadblocksizestreamidzenableonstart 0conckerneltrace	key_valuec                    t                      }t          |d          st          d          t          t          j        d          r[t          j        j        Jt          t          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 }|                    d                    d |D                                  |                                 t%          |                    |j        | |                     d d d            d S # 1 swxY w Y   d S )NcudaOutputModez-HIP does not support profiler initialization!cuda.r      z.CUDA12+ does not need profiler initialization!r   csvz;supported CUDA profiler output modes are: key_value and csvT)delete   
c              3   @   K   | ]}|                     d           V  dS )asciiN)encode).0fs     S/var/www/html/auto_sub_bot/venv/lib/python3.11/site-packages/torch/cuda/profiler.py	<genexpr>zinit.<locals>.<genexpr>*   s.      <<188G,,<<<<<<    )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'(( NLMMMv&&OM*"((--a011R77 MNNN"]MMEk!!,9			,0I
 
 	
 
	$D	1	1	1 VQ	

<<e<<<<<===				B--afkCSTTUUUV V V V V V V V V V V V V V V V V Vs   8A0E55E9<E9c                  `    t          t                                                                 d S N)r   r   cudaProfilerStart r    r   r   r   /   s&    **,,-----r    c                  `    t          t                                                                 d S r8   )r   r   cudaProfilerStopr:   r    r   r   r   3   s&    ))++,,,,,r    c               #   r   K   	 t                       d V  t                       d S # t                       w xY wr8   )r   r   r:   r    r   r	   r	   7   s8      s   & 6)Nr   )
contextlibr+   r#    r   r   __all__r'   r   r   r   contextmanagerr	   r:   r    r   <module>rB      s          ! ! ! ! ! ! ! !
.
.
.  V V V V6. . .- - -     r    