NVMLDevicePool.hpp

Go to the documentation of this file.

/*
 * Copyright (c) 2015 - 2024 Intel Corporation
 * SPDX-License-Identifier: BSD-3-Clause
 */
 
#ifndef NVMLDEVICEPOOL_HPP_INCLUDE
#define NVMLDEVICEPOOL_HPP_INCLUDE
 
#include <vector>
#include <string>
#include <cstdint>
 
#include "geopm_sched.h"
 
namespace geopm
{
 
    class NVMLDevicePool
    {
        public:
            NVMLDevicePool() = default;
            virtual ~NVMLDevicePool() = default;
            virtual int num_gpu(void) const = 0;
            virtual std::unique_ptr<cpu_set_t, std::function<void(cpu_set_t *)> >
                cpu_affinity_ideal_mask(int gpu_idx) const = 0;
            virtual uint64_t frequency_status_sm(int gpu_idx) const = 0;
            virtual std::vector<unsigned int> frequency_supported_sm(int gpu_idx) const = 0;
            virtual uint64_t utilization(int gpu_idx) const = 0;
            virtual uint64_t power (int gpu_idx) const = 0;
            virtual uint64_t power_limit (int gpu_idx) const = 0;
            virtual uint64_t frequency_status_mem(int gpu_idx) const = 0;
            virtual uint64_t throttle_reasons(int gpu_idx) const = 0;
            virtual uint64_t temperature(int gpu_idx) const = 0;
            virtual uint64_t energy(int gpu_idx) const = 0;
            virtual uint64_t performance_state(int gpu_idx) const = 0;
            virtual uint64_t throughput_rx_pcie(int gpu_idx) const = 0;
            virtual uint64_t throughput_tx_pcie(int gpu_idx) const = 0;
            virtual uint64_t utilization_mem(int gpu_idx) const = 0;
            virtual std::vector<int> active_process_list(int gpu_idx) const = 0;
 
            virtual void frequency_control_sm(int gpu_idx, int min_freq, int max_freq) const = 0;
            virtual void frequency_reset_control(int gpu_idx) const = 0;
            virtual void power_control(int gpu_idx, int setting) const = 0;
 
            virtual bool is_privileged_access(void) const = 0;
            virtual void reset(void) = 0;
        private:
    };
 
    NVMLDevicePool &nvml_device_pool(int num_cpu);
}
#endif