
    .`i                        d dl Z d dlZd dlmZ d dlmZ  ee          Z ed          dup ed          duZer 	 d dl	m
Z
 d  e
j                    D             Ze j                            d          Zeduo  ee                                          d k    Zer% ee          d k    re                    d	           n2es0 ee          d
k    r!e                    d ee                     dZnG# e$ r e                    d           dZY n)e$ r"Ze                    de           dZY dZ[ndZ[ww xY wese                    d            G d dej                  Z G d dej                  ZdS )    N)	find_spec)init_loggertritonzpytorch-triton-xpu)backendsc                 \    g | ])}|j         	|j                                         "|j         *S  )driver	is_active).0xs     o/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/triton_utils/importing.py
<listcomp>r      sG     
 
 
18
@A@R@R@T@T
H
 
 
    CUDA_VISIBLE_DEVICESzaTriton found 0 active drivers in distributed environment. This is expected during initialization.   zkTriton is installed but %d active driver(s) found (expected 1). Disabling Triton to prevent runtime errors.FzSTriton is installed, but `triton.backends` could not be imported. Disabling Triton.zXAn unexpected error occurred while checking Triton active drivers: %s. Disabling Triton.z\Triton not installed or not compatible; certain GPU-related functions will not be available.c                   $     e Zd Z fdZd Z xZS )TritonPlaceholderc                 N   t                                          d           d| _        |                     d          | _        |                     d          | _        |                     d          | _        |                     d          | _        t                      | _	        d S )Nr   z3.4.0jitautotune
heuristicsConfig)
super__init____version___dummy_decoratorr   r   r   r   TritonLanguagePlaceholderlanguageself	__class__s    r   r   zTritonPlaceholder.__init__K   s    """"((//--j99//==++H55133r   c                     d }|S )Nc                  F    | rt          | d                   r| d         S d S )Nr   c                     | S )Nr   )fs    r   <lambda>zGTritonPlaceholder._dummy_decorator.<locals>.decorator.<locals>.<lambda>X   s    Q r   )callable)argskwargss     r   	decoratorz5TritonPlaceholder._dummy_decorator.<locals>.decoratorU   s.     a)) Aw;r   r   )r    namer*   s      r   r   z"TritonPlaceholder._dummy_decoratorT   s    	 	 	
 r   )__name__
__module____qualname__r   r   __classcell__r!   s   @r   r   r   J   sG        4 4 4 4 4      r   r   c                        e Zd Z fdZ xZS )r   c                     t                                          d           d | _        d | _        d | _        d | _        d | _        d | _        d | _        d | _	        d S )Nztriton.language)
r   r   	constexprdtypeint64int32tensorexploglog2r   s    r   r   z"TritonLanguagePlaceholder.__init__^   sX    *+++


			r   )r,   r-   r.   r   r/   r0   s   @r   r   r   ]   s8        	 	 	 	 	 	 	 	 	r   r   )ostypesimportlib.utilr   vllm.loggerr   r,   logger
HAS_TRITONtriton.backendsr   valuesactive_driversenvirongetcuda_visible_deviceslenstripis_distributed_envdebuginfoImportErrorwarning	Exceptione
ModuleTyper   r   r   r   r   <module>rQ      s   
			  $ $ $ $ $ $ # # # # # #	X		 Iht# 3y%&&d2   10,,,,,,

 
-ho//
 
 
  "z~~.DEE ,W5I5O5O5Q5Q1R1RVW1W 	
  	##n"5"5":": LL:    $ 	N(;(;q(@(@KKLN##  
 J    	 	
 	
 	
 


   %	
 	
 	

 





  
KK	,      (   &
 
 
 
 
 0 
 
 
 
 
s   B7C6 6D:D:D55D: