U
    T?hz                     @   sJ   d dl Z d dlZe eZG dd dZedkrFe Ze  e  dS )    Nc                   @   s$   e Zd Zdd Zdd Zdd ZdS )AffinitySettingc                 C   s:   t  | _d | _tt do"tt d| _| js6td 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self r   Z/var/www/html/venv/lib/python3.8/site-packages/onnxruntime/transformers/affinity_helper.py__init__   s
    
zAffinitySetting.__init__c                 C   s   | j rt| j| _d S )N)r
   r   r   r   r   r   r   r   r   get_affinity   s    zAffinitySetting.get_affinityc                 C   sH   | j rDt| j}| j|krDtdt|t| j t| j| j d S )Nz%Replacing affinity setting %s with %s)	r
   r   r   r   r   r   r   strr   )r   Zcurrent_affinityr   r   r   set_affinity   s    
zAffinitySetting.set_affinityN)__name__
__module____qualname__r   r   r   r   r   r   r   r      s   r   __main__)	loggingr   	getLoggerr   r   r   Zaffi_helperr   r   r   r   r   r   <module>   s   
