3
�<�es � @ sj d dl mZ d dlT ddlZddlmZ ddlZddlZej j
� 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 � Ze d d
� �Z
dd� Z e ddd�dd� �Z
ed�d dd��Ze ddd�dd� �Zed�d dd ��Zd S ) � UncorePlugina|
`uncore`::
`max_freq_khz, min_freq_khz`:::
Limit the maximum and minumum uncore frequency.
Those options are Intel specific and correspond directly to `sysfs` files
exposed by Intel uncore frequency driver.
====
----
[uncore]
max_freq_khz=4000000
----
Using this options *TuneD* will limit maximum frequency of all uncore units
on the Intel system to 4 GHz.
====
c C s� d| _ t� | _t� | _d| _ytjt�}W n tk
r> d S X t j
|d�}t |�dkrbd| _|}x|D ]}| jj |� qhW t
jdt| 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.6/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_init: s zUncorePlugin._instance_initc C s d S )Nr )r r r r r �_instance_cleanup> s z UncorePlugin._instance_cleanupc C s2 t | d | }tj|�}t|�dkr.t|�S d S )N�/r )r
�cmdZ read_filer �int)r �dev_dir�file�
sysfs_file�valuer r r �_getA s
zUncorePlugin._getc C s( t | d | }tj|d| �r$|S d S )Nr z%u)r
r Z
write_to_file)r r r r! r r r r �_setH s zUncorePlugin._setc C s
d d d�S )N)� max_freq_khz� min_freq_khzr )�clsr r r �_get_config_optionsN s z UncorePlugin._get_config_optionsc C s* y t |�}W n" tk
r. tjd| � d S X y4| j|d�}| j|d�}| j|d�}| j|d�}W n" ttfk
r� tjd� d S X |tkr�||k r�tjd|||f � d S ||kr�tjd|||f � |}nT|t k�r"||k� r�tjd |||f � d S ||k �r&tjd
|||f � |}nd S |S ) Nzvalue '%s' is not integer�initial_max_freq_khz�initial_min_freq_khzr$ r% z+fail to read inital uncore frequency valuesz/%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 �errorr" r �IOError�IS_MAX�info�IS_MIN) r �deviceZ
min_or_maxr! Zfreq_khzr( r) r$ r% r r r �_validate_valueU s:
z UncorePlugin._validate_valuer$ T)Z
per_devicec C sB | j |t|�}|d krd S |r"|S tjd||f � | j|d|�S )Nz%s: set max_freq_khz %dr$ )r1 r- r r r# )r r! r0 �sim�remover$ r r r �_set_max_freq_khz| s z UncorePlugin._set_max_freq_khzFc C s` |rt jjt� rd S y| j|d�}W n" ttfk
rH tjd� d S X tj d||f � |S )Nr$ z$fail to read uncore frequency valuesz%s: get max_freq_khz %d)
r �path�isdirr
r" r r, r r+ r )r r0 �ignore_missingr$ r r r �_get_max_freq_khz� s
z UncorePlugin._get_max_freq_khzr% c C sB | j |t|�}|d krd S |r"|S tjd||f � | j|d|�S )Nz%s: set min_freq_khz %dr% )r1 r/ r r r# )r r! r0 r2 r3 r% r r r �_set_min_freq_khz� s z UncorePlugin._set_min_freq_khzc C s` |rt jjt� rd S y| j|d�}W n" ttfk
rH tjd� d S X tj d||f � |S )Nr% z$fail to read uncore frequency valuesz%s: get min_freq_khz %d)
r r5 r6 r
r" r r, r r+ r )r r0 r7 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# � classmethodr' r1 Z command_setr4 Z command_getr8 r9 r: r r r r r s '
r )� r Z
decoratorsZ
tuned.logsZtunedZtuned.utils.commandsr r r Zlogs�getr r r
r/ r- ZPluginr r r r r �<module> s