
    wi                     l    d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZe j        d             Z	d Z
dS )    Nc               #   (  K   t           j        t           j        }} t          j                    t          j                    ct           _        t           _        	 d V  | |ct           _        t           _        d S # | |ct           _        t           _        w xY wN)sysstdoutstderrioStringIO)
old_stdout
old_stderrs     d/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/triton/runtime/build.pyquietr      sm       Z
J[]]BKMMCJ
8!+Z
CJJJZ
CJ7777s   A6 6Bc           
      n   t          j        d          }t          j                            |d                    | |                    }t          j                            d          }|?t          j	        d          }	t          j	        d          }
|
|
n|	}|t          d          t          t           d          rt          j                    }nt          j                    }|d	k    rd
}t          j        |          d         }t          d dD                       }|||g|z   }||ddddd|g}|d |D             z  }|d |D             z  }|d |D             z  }t!          j        |          }|dk    r|S g }g }t%          j        | d|g||dgz   |||          }dg}|                    d|z              |                    d|z              |                    d           t+          | |g|          }t-                      5  t%          j        di | d d d            n# 1 swxY w Y   |S ) N
EXT_SUFFIXz{name}{suffix})namesuffixCCclanggcczFFailed to find C compiler. Please specify via CC environment variable.get_default_schemeposix_localposix_prefix)schemeincludec              3   >   K   | ]}t          j        |          V  d S r   )osgetenv).0vars     r   	<genexpr>z_build.<locals>.<genexpr>+   s*      ffbinnffffff    )TRITON_CUDACRT_PATHTRITON_CUDART_PATHz-O3z-sharedz-fPICz
-Wno-psabiz-oc                     g | ]}d | S )z-l )r   libs     r   
<listcomp>z_build.<locals>.<listcomp>/   s    ///czCzz///r    c                     g | ]}d | S )z-Lr$   r   dirs     r   r&   z_build.<locals>.<listcomp>0   s    222czCzz222r    c                     g | ]	}|d| 
S )Nz-Ir$   r(   s     r   r&   z_build.<locals>.<listcomp>1   s    EEEcS_zCzz___r    r   c)r   languagesourcesinclude_dirsextra_compile_argsextra_link_argslibrary_dirs	libraries	build_extz--build-temp=z--build-lib=z-q)r   ext_modulesscript_argsr$   )	sysconfigget_config_varr   pathjoinformatenvirongetshutilwhichRuntimeErrorhasattrr   _get_default_scheme	get_pathsset
subprocess
check_call
setuptools	Extensionappenddictr   setup)r   srcsrcdirr1   r.   r2   r   soccr   r   r   py_include_dircustom_backend_dirscc_cmdretr/   r0   extargss                       r   _buildrU      s   %l33F	f.5545OO	P	PB			B	zW%%l5!!OSS:ghhhy.// 1-//.00 (777	BNff8efffff6>"P<O"PPL#ui,bIF
//Y////F
22\2222F
EE\EEEEF


'
'C
axx	O

!-7'!	 	 	C =DKK&()))KK'(((KKE  D
 
 ! !  4   ! ! ! ! ! ! ! ! ! ! ! ! ! ! !Is   H**H.1H.)
contextlibr   r   r6   r   r=   rD   rF   contextmanagerr   rU   r$   r    r   <module>rX      s        



 				     				          8 8 8; ; ; ; ;r    