
    hz                         S SK r S SKr\ R                  " \5      r " S S5      r\S:X  a(  \" 5       r\R                  5         \R                  5         gg)    Nc                   &    \ rS rSrS rS rS rSrg)AffinitySetting   c                     [         R                  " 5       U l        S U l        [	        [         S5      =(       a    [	        [         S5      U l        U R
                  (       d  [        R                  S5        g g )Nsched_getaffinitysched_setaffinityzCCurrent OS does not support os.get_affinity() and os.set_affinity())osgetpidpidaffinityhasattris_os_supportedloggerwarningselfs    b/var/www/fran/franai/venv/lib/python3.13/site-packages/onnxruntime/transformers/affinity_helper.py__init__AffinitySetting.__init__   sN    99;&r+>?dGBPcDd##NN`a $    c                 r    U R                   (       a&  [        R                  " U R                  5      U l        g g )N)r   r	   r   r   r   r   s    r   get_affinityAffinitySetting.get_affinity   s&    00:DM  r   c                 F   U R                   (       a  [        R                  " U R                  5      nU R                  U:w  a_  [
        R                  S[        U5      [        U R                  5      5        [        R                  " U R                  U R                  5        g g g )Nz%Replacing affinity setting %s with %s)	r   r	   r   r   r   r   r   strr   )r   current_affinitys     r   set_affinityAffinitySetting.set_affinity   su    !33DHH=}} 00;()&
 $$TXXt}}= 1  r   )r   r   r   N)__name__
__module____qualname____firstlineno__r   r   r   __static_attributes__ r   r   r   r      s    b;	>r   r   __main__)	loggingr	   	getLoggerr   r   r   affi_helperr   r   r$   r   r   <module>r)      sV     				8	$> >0 z!#K r   