a
���h� � @ sj d dl mZ d dlT ddlZddlmZ ddlZddlZej �
� Z e� Z dZ
dZd ZG dd� dej�ZdS ) � )�hotplug)�*� N)�commandsz//sys/devices/system/cpu/intel_uncore_frequency/c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z e
dd� �Z dd� Z dd� Z
dd� Zeddd�dd� �Zed�d$d d ��Zed dd�d d � �Zed �d%d!d"��Zd#S )&� UncorePlugina�
An Intel-specific plug-in for limiting the maximum and minimum uncore frequency.
The options [option]`max_freq_khz`, [option]`min_freq_khz` correspond to
`sysfs` files exposed by Intel uncore frequency driver. Their values can be
specified in kHz or as a percentage of their configurable range.
.Limiting maximum uncore frequency
====
----
[uncore10]
type=uncore
devices=uncore10
max_freq_khz=4000000
[uncore_all]
type=uncore
max_freq_khz=90%
----
Using this options *TuneD* will limit maximum frequency of all uncore units
on the Intel system to 90% of the allowable range. Except uncore10 which
maximum frequency limit will be set to 4 GHz.
====
c C s� d| _ t� | _t� | _d| _zt�t�}W n ty> Y d S 0 t �
|d�}t |�dkrbd| _|}|D ]}| j� |� qft
�dt| j�� d S )NTFzuncore*r z devices: %s)Z_devices_supported�setZ_assigned_devicesZ
_free_devicesZ_is_tpmi�os�listdir� SYSFS_DIR�OSError�fnmatch�filter�len�add�log�debug�str)�selfZdevicesZ tpmi_devices�d� r �?/usr/lib/python3.9/site-packages/tuned/plugins/plugin_uncore.py�
_init_devices+ s zUncorePlugin._init_devicesc C s d|_ d|_d S )NTF)Z_has_static_tuningZ_has_dynamic_tuning�r �instancer r r �_instance_initA s zUncorePlugin._instance_initc C s d S )Nr r r r r �_instance_cleanupE s z UncorePlugin._instance_cleanupc C s2 t | d | }t�|�}t|�dkr.t|�S d S )N�/r )r
�cmdZ read_filer �int)r �dev_dir�file�
sysfs_file�valuer r r �_getH s
zUncorePlugin._getc C s( t | d | }t�|d| �r$|S d S )Nr z%u)r
r Z
write_to_file)r r r r" r! r r r �_setO s zUncorePlugin._setc C sd z4| � |d�}| � |d�}| � |d�}| � |d�}W n" ttfyV t�d� Y d S 0 ||||fS )N�initial_max_freq_khz�initial_min_freq_khz� max_freq_khz� min_freq_khz�$fail to read uncore frequency values)r# r �IOErrorr �error)r �devicer% r& r' r( r r r �_get_allU s
zUncorePlugin._get_allc C s
d d d�S )N)r' r( r )�clsr r r �_get_config_options` s �z UncorePlugin._get_config_optionsc
C s� z t |�}W n" ty. t�d| � Y d S 0 | �|�}|d u rFd S |\}}}} |tkr�|| k rzt�d||| f � d S ||kr�t�d|||f � |}nN|tkr�||kr�t�d|||f � d S ||k r�t�d|||f � |}nd S |S )N�value '%s' is not integerz/%s: max_freq_khz %d value below min_freq_khz %dzC%s: max_freq_khz %d value above initial_max_freq_khz - capped to %dz/%s: min_freq_khz %d value above max_freq_khz %dzC%s: min_freq_khz %d value below initial_max_freq_khz - capped to %d)r �
ValueErrorr r+ r- �IS_MAX�info�IS_MIN)
r r, �
min_or_maxr" Zfreq_khz�valuesr% r& r' r( r r r �_validate_khz_valueg s2
z UncorePlugin._validate_khz_valuec C sV z t |�}W n" ty. t�d| � Y d S 0 |dk s@|dkrRt�d| � d S |S )Nr0 r �d z/percent value '%s' is not within [0..100] range)r r1 r r+ )r r" �pctr r r �_validate_percent_value� s z$UncorePlugin._validate_percent_valuec
C s� t |t�rn|d dkrn| �|�d��}|d u r2d S | �|�}|d u rHd S |\}}}}|t||| d � } n|} | �||| �S )N����%r8 )�
isinstancer r: �rstripr- r r7 )
r r, r5 r" r9 r6 r% r& �_Zkhzr r r �_validate_value� s
z UncorePlugin._validate_valuer' T)Z
per_devicec C sB | � |t|�}|d u rd S |r"|S t�d||f � | �|d|�S )Nz%s: set max_freq_khz %dr' )r@ r2 r r r$ )r r" r, r �sim�remover' r r r �_set_max_freq_khz� s z UncorePlugin._set_max_freq_khzFc C s^ |rt j�t�sd S z| �|d�}W n" ttfyF t�d� Y d S 0 t� d||f � |S )Nr' r) z%s: get max_freq_khz %d�
r �path�isdirr
r# r r* r r+ r )r r, r �ignore_missingr' r r r �_get_max_freq_khz� s
z UncorePlugin._get_max_freq_khzr( c C sB | � |t|�}|d u rd S |r"|S t�d||f � | �|d|�S )Nz%s: set min_freq_khz %dr( )r@ r4 r r r$ )r r" r, r rA rB r( r r r �_set_min_freq_khz� s z UncorePlugin._set_min_freq_khzc C s^ |rt j�t�sd S z| �|d�}W n" ttfyF t�d� Y d S 0 t� d||f � |S )Nr( r) z%s: get min_freq_khz %drD )r r, r rG r( r r r �_get_min_freq_khz� s
z UncorePlugin._get_min_freq_khzN)F)F)�__name__�
__module__� __qualname__�__doc__r r r r# r$ r- � classmethodr/ r7 r: r@ Z command_setrC Z command_getrH rI rJ r r r r r s(
#
r )� r Z
decoratorsZ
tuned.logsZtunedZtuned.utils.commandsr r r Zlogs�getr r r
r4 r2 ZPluginr r r r r �<module> s