o
    
jz                     @   sN   d dl Z d dlZe eZG dd dZedkr%e Ze  e  dS 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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self r   [/home/kuhnn/.local/lib/python3.10/site-packages/onnxruntime/transformers/affinity_helper.py__init__   s   
zAffinitySetting.__init__c                 C   s   | j rt| j| _d S d S )N)r
   r   r   r   r   r   r   r   r   get_affinity   s   zAffinitySetting.get_affinityc                 C   sP   | j r$t| j}| j|kr&tdt|t| j t| 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_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   affi_helperr   r   r   r   r   r   <module>   s   
