
    gz                         d dl Z d dlZ e j                  e      Z G d d      Zedk(  r( e       Zej                          ej                          yy)    Nc                       e Zd Zd Zd Zd Zy)AffinitySettingc                     t        j                         | _        d | _        t	        t         d      xr t	        t         d      | _        | j
                  st        j                  d       y y )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    ]/var/www/openai/venv/lib/python3.12/site-packages/onnxruntime/transformers/affinity_helper.py__init__zAffinitySetting.__init__   sN    99;&r+>?dGBPcDd##NN`a $    c                 f    | j                   r%t        j                  | j                        | _        y y )N)r   r   r   r
   r   r   s    r   get_affinityzAffinitySetting.get_affinity   s&    00:DM  r   c                 6   | j                   rt        j                  | j                        }| j                  |k7  r^t
        j                  dt        |      t        | j                               t        j                  | j                  | j                         y y y )Nz%Replacing affinity setting %s with %s)	r   r   r   r
   r   r   r   strr   )r   current_affinitys     r   set_affinityzAffinitySetting.set_affinity   su    !33DHH=}} 00;()&
 $$TXXt}}= 1  r   N)__name__
__module____qualname__r   r   r    r   r   r   r      s    b;	>r   r   __main__)	loggingr   	getLoggerr   r   r   affi_helperr   r   r   r   r   <module>r#      sX     				8	$> >0 z!#K r   