Subversion Repositories ALCASAR

<?php
/**
 * nvidiasmi sensor class, getting hardware temperature information and fan speed from nvidia-smi utility
 *
 * PHP version 5
 *
 * @category  PHP
 * @package   PSI_Sensor
 * @author    Mieczyslaw Nalewaj <namiltd@users.sourceforge.net>
 * @copyright 2020 phpSysInfo
 * @license   http://opensource.org/licenses/gpl-2.0.php GNU General Public License version 2, or (at your option) any later version
 * @version   Release: 3.0
 * @link      http://phpsysinfo.sourceforge.net
 */
class NvidiaSMI extends Sensors
{
    /**
     * content to parse
     *
     * @var array
     */
    private $_gpus = array();

    /**
     * fill the private array
     */
    public function __construct()
    {
        parent::__construct();
        if (!defined('PSI_EMU_HOSTNAME')) switch (defined('PSI_SENSOR_NVIDIASMI_ACCESS')?strtolower(PSI_SENSOR_NVIDIASMI_ACCESS):'command') {
        case 'command':
            if (PSI_OS == 'WINNT') {
                $winnt_exe = (defined('PSI_SENSOR_NVIDIASMI_EXE_PATH') && is_string(PSI_SENSOR_NVIDIASMI_EXE_PATH))?strtolower(PSI_SENSOR_NVIDIASMI_EXE_PATH):"c:\\Program Files\\NVIDIA Corporation\\NVSMI\\nvidia-smi.exe";
                if (($_exe=realpath(trim($winnt_exe))) && preg_match("/^([a-zA-Z]:\\\\[^\\\\]+)/", $_exe, $out)) {
                    CommonFunctions::executeProgram('cmd', "/c set ProgramFiles=".$out[1]."^&\"".$_exe."\" -q", $lines);
                } else {
                    $this->error->addConfigError('__construct()', '[sensor_nvidiasmi] EXE_PATH="'.$winnt_exe.'"');
                }
            } else {
                CommonFunctions::executeProgram('nvidia-smi', '-q', $lines);
            }

            $this->_gpus = preg_split("/^(?=GPU )/m", $lines, -1, PREG_SPLIT_NO_EMPTY);
            break;
        case 'data':
            if (CommonFunctions::rfts(PSI_APP_ROOT.'/data/nvidiasmi.txt', $lines)) {
                $this->_gpus = preg_split("/^(?=GPU )/m", $lines, -1, PREG_SPLIT_NO_EMPTY);
            }
            break;
        default:
            $this->error->addConfigError('__construct()', '[sensor_nvidiasmi] ACCESS');
            break;
        }
    }

    /**
     * get the information
     *
     * @see PSI_Interface_Sensor::build()
     *
     * @return Void
     */
    public function build()
    {
        $gpuc=count($this->_gpus);
        switch ($gpuc) {
            case 0:
                $this->error->addError("nvidia-smi", "No values");
                break;
            case 1:
                $this->error->addError("nvidia-smi", "Error: ".$this->_gpus[0]);
                break;
            default:
                for ($c = 0; $c < $gpuc; $c++) {
                    if (preg_match("/^\s+GPU Current Temp\s+:\s*(\d+)\s*C\s*$/m", $this->_gpus[$c], $out)) {
                        $dev = new SensorDevice();
                        $dev->setName("GPU ".($c)." (nvidiasmi)");
                        $dev->setValue($out[1]);
                        if (preg_match("/^\s+GPU Shutdown Temp\s+:\s*(\d+)\s*C\s*$/m", $this->_gpus[$c], $out)) {
                            $dev->setMax($out[1]);
                        }
                        $this->mbinfo->setMbTemp($dev);
                    }
                    if (preg_match("/^\s+Power Draw\s+:\s*([\d\.]+)\s*W\s*$/m", $this->_gpus[$c], $out)) {
                        $dev = new SensorDevice();
                        $dev->setName("GPU ".($c)." (nvidiasmi)");
                        $dev->setValue($out[1]);
                        if (preg_match("/^\s+Power Limit\s+:\s*([\d\.]+)\s*W\s*$/m", $this->_gpus[$c], $out)) {
                            $dev->setMax($out[1]);
                        }
                        $this->mbinfo->setMbPower($dev);
                    }
                    if (preg_match("/^\s+Fan Speed\s+:\s*(\d+)\s*%\s*$/m", $this->_gpus[$c], $out)) {
                        $dev = new SensorDevice();
                        $dev->setName("GPU ".($c)." (nvidiasmi)");
                        $dev->setValue($out[1]);
                        $dev->setUnit("%");
                        $this->mbinfo->setMbFan($dev);
                    }
                    if (preg_match("/^\s+Performance State\s+:\s*(\S+)\s*$/m", $this->_gpus[$c], $out)) {
                        $dev = new SensorDevice();
                        $dev->setName("GPU ".($c)." Performance State (nvidiasmi)");
                        $dev->setValue($out[1]);
                        $this->mbinfo->setMbOther($dev);
                    }
                    if (preg_match("/^\s+Gpu\s+:\s*(\d+)\s*%\s*$/m", $this->_gpus[$c], $out)) {
                        $dev = new SensorDevice();
                        $dev->setName("GPU ".($c)." Utilization (nvidiasmi)");
                        $dev->setValue($out[1]);
                        $dev->setUnit("%");
                        $this->mbinfo->setMbOther($dev);
                    }
                    if (preg_match("/^\s+Memory\s+:\s*(\d+)\s*%\s*$/m", $this->_gpus[$c], $out)) {
                        $dev = new SensorDevice();
                        $dev->setName("GPU ".($c)." Memory Utilization (nvidiasmi)");
                        $dev->setValue($out[1]);
                        $dev->setUnit("%");
                        $this->mbinfo->setMbOther($dev);
                    }
                    if (preg_match("/^\s+Encoder\s+:\s*(\d+)\s*%\s*$/m", $this->_gpus[$c], $out)) {
                        $dev = new SensorDevice();
                        $dev->setName("GPU ".($c)." Encoder Utilization (nvidiasmi)");
                        $dev->setValue($out[1]);
                        $dev->setUnit("%");
                        $this->mbinfo->setMbOther($dev);
                    }
                    if (preg_match("/^\s+Decoder\s+:\s*(\d+)\s*%\s*$/m", $this->_gpus[$c], $out)) {
                        $dev = new SensorDevice();
                        $dev->setName("GPU ".($c)." Decoder Utilization (nvidiasmi)");
                        $dev->setValue($out[1]);
                        $dev->setUnit("%");
                        $this->mbinfo->setMbOther($dev);
                    }
                }
                break;
        }
    }
}