
     Ngz                         d dl Z d dlZ e j        e          Z G d d          Zedk    r4 e            Ze                                 e                                 dS dS )    Nc                        e Zd Zd Zd Zd ZdS )AffinitySettingc                     t          j                    | _        d | _        t	          t           d          ot	          t           d          | _        | j        st                              d           d S d S )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    d/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnxruntime/transformers/affinity_helper.py__init__zAffinitySetting.__init__   sj    9;;&r+>??dGBPcDdDd# 	bNN`aaaaa	b 	b    c                 T    | j         r t          j        | j                  | _        d S d S )N)r   r   r   r
   r   r   s    r   get_affinityzAffinitySetting.get_affinity   s/     	;0::DMMM	; 	;r   c                    | j         rt          j        | j                  }| j        |k    r^t
                              dt          |          t          | j                             t          j        | j        | j                   d S d S d S )Nz%Replacing affinity setting %s with %s)	r   r   r   r
   r   r   r   strr   )r   current_affinitys     r   set_affinityzAffinitySetting.set_affinity   s     	>!3DH==} 000;())&&  
 $TXt}=====	> 	>00r   N)__name__
__module____qualname__r   r   r    r   r   r   r      sD        b b b; ; ;	> 	> 	> 	> 	>r   r   __main__)	loggingr   	getLoggerr   r   r   affi_helperr   r   r   r   r   <module>r#      s     						8	$	$> > > > > > > >0 z!/##K r   