3
ܓi @ s& d dl mZmZ G dd deeZdS ) )PluginIndependentPluginc @ s0 e Zd ZdZdZdZd
Zedd Zdd Z d S ) Spyreu Spyre chip is IBM’s AI accelerator, designed to handle AI inferencing
and workloads.
The Spyre plugin collects data about the Spyre card’s VFIO device node
tree, configuration files, and more.
z!IBM Spyre Accelerator InformationZspyreppc.*c C s g }| s dS x| j D ]}|j s$qy*|jdd\}}}}|j jd\}}W n tk
rd wY nX |dkr|d kr|jddkrd | }y"|jd\} }
} | jd
\} }
W n tk
r wY nX |j| |
| |
f qW |S ) a Extract PCI domain, bus, device, function for devices that match:
- Vendor ID = 0x1014 (IBM)
- Device ID = 0x06a7 or 0x06a8
Parameters
----------
lspci_out : str
The output string from `lspci -n`.
Returns
-------
list of tuples
A list of (domain, bus, device, function) tuples for each matching
card.
N )maxsplit:Z101406a706a8 z0000:.)r r
)
splitlinesstripsplitlower
ValueErrorcountappend)Z lspci_output spyre_cardslinepci_addrZ_classZidsZ_restZvendordevicedomainbusZdev_funcZ
pci_devicefunction r /usr/lib/python3.6/spyre.pyget_ibm_spyre_devices s* zSpyre.get_ibm_spyre_devicesc C s | j d}|d dkrd S | j|d }|s0d S xT|D ]L\}}}}| d| d| d| }d| d| d| d }| j|d
d q6W | jd d
dg d S )Nzlspci -nZstatusr outputr r z/sys/devices/pci/z /vfio-devT)Ztreez /etc/modprobe.d/vfio-pci.confz!/etc/udev/rules.d/95-vfio-3.rulesz#/etc/security/limits.d/memlock.conf)Zexec_cmdr Zadd_dir_listingZ
add_copy_spec) selfZlspcir r r r r r Zpci_vfior r r setupH s
z Spyre.setupN)r )
__name__
__module__ __qualname____doc__Z
short_descZ plugin_nameZ
architectures staticmethodr r! r r r r r s .r N)Zsos.report.pluginsr r r r r r r s