NVMLIOGroup.hpp

Go to the documentation of this file.

/*
 * Copyright (c) 2015 - 2024 Intel Corporation
 * SPDX-License-Identifier: BSD-3-Clause
 */
 
#ifndef NVMLIOGROUP_HPP_INCLUDE
#define NVMLIOGROUP_HPP_INCLUDE
 
#include <map>
#include <vector>
#include <string>
#include <memory>
 
#include "geopm/IOGroup.hpp"
 
namespace geopm
{
    class PlatformTopo;
    class NVMLDevicePool;
    class SaveControl;
 
    class NVMLIOGroup : public IOGroup
    {
        public:
            NVMLIOGroup();
            NVMLIOGroup(const PlatformTopo &platform_topo,
                        const NVMLDevicePool &device_pool,
                        std::shared_ptr<SaveControl> save_control);
            virtual ~NVMLIOGroup() = default;
            std::set<std::string> signal_names(void) const override;
            std::set<std::string> control_names(void) const override;
            bool is_valid_signal(const std::string &signal_name) const override;
            bool is_valid_control(const std::string &control_name) const override;
            int signal_domain_type(const std::string &signal_name) const override;
            int control_domain_type(const std::string &control_name) const override;
            int push_signal(const std::string &signal_name, int domain_type, int domain_idx)  override;
            int push_control(const std::string &control_name, int domain_type, int domain_idx) override;
            void read_batch(void) override;
            void write_batch(void) override;
            double sample(int batch_idx) override;
            void adjust(int batch_idx, double setting) override;
            double read_signal(const std::string &signal_name, int domain_type, int domain_idx) override;
            void write_control(const std::string &control_name, int domain_type, int domain_idx, double setting) override;
            void save_control(void) override;
            void restore_control(void) override;
            std::function<double(const std::vector<double> &)> agg_function(const std::string &signal_name) const override;
            std::function<std::string(double)> format_function(const std::string &signal_name) const override;
            std::string signal_description(const std::string &signal_name) const override;
            std::string control_description(const std::string &control_name) const override;
            int signal_behavior(const std::string &signal_name) const override;
            void save_control(const std::string &save_path) override;
            void restore_control(const std::string &save_path) override;
            std::string name(void) const override;
            static std::string plugin_name(void);
            static std::unique_ptr<geopm::IOGroup> make_plugin(void);
        private:
            void register_signal_alias(const std::string &alias_name, const std::string &signal_name);
            void register_control_alias(const std::string &alias_name, const std::string &control_name);
 
            std::map<pid_t, double> gpu_process_map(void) const;
            double cpu_gpu_affinity(int cpu_idx, std::map<pid_t, double> process_map) const;
 
            static const std::string M_PLUGIN_NAME;
            static const std::string M_NAME_PREFIX;
            const PlatformTopo &m_platform_topo;
            const NVMLDevicePool &m_nvml_device_pool;
            bool m_is_batch_read;
            std::vector<double> m_frequency_max_control_request;
            std::vector<double> m_frequency_min_control_request;
            std::vector<double> m_initial_power_limit;
            std::vector<std::vector<unsigned int> > m_supported_freq;
            std::vector<double> m_frequency_step;
 
            struct signal_s
            {
                double m_value;
                bool m_do_read;
            };
 
            struct control_s
            {
                double m_setting;
                bool m_is_adjusted;
            };
 
            struct signal_info {
                std::string m_description;
                std::vector<std::shared_ptr<signal_s> > signals;
                int domain;
                std::function<double(const std::vector<double> &)> agg_function;
                int behavior;
                std::function<std::string(double)> format_function;
            };
 
            struct control_info {
                std::string m_description;
                std::vector<std::shared_ptr<control_s> > controls;
                int domain;
                std::function<double(const std::vector<double> &)> agg_function;
                std::function<std::string(double)> format_function;
            };
 
            std::map<std::string, signal_info> m_signal_available;
            std::map<std::string, control_info> m_control_available;
            std::vector<std::shared_ptr<signal_s> > m_signal_pushed;
            std::vector<std::shared_ptr<control_s> > m_control_pushed;
 
            std::shared_ptr<SaveControl> m_mock_save_ctl;
    };
}
#endif