blob: 0a85b9d6a80ef626fecb4b5366227718f6ce6391 [file] [log] [blame]
Harshit Agherafa2a5b92025-05-22 11:35:39 +05301/*
2 * SPDX-FileCopyrightText: Copyright (c) 2024-2025 NVIDIA CORPORATION &
3 * AFFILIATES. All rights reserved.
4 * SPDX-License-Identifier: Apache-2.0
5 */
6
7#pragma once
8
9#include "MctpRequester.hpp"
10#include "NvidiaDeviceDiscovery.hpp"
Harshit Agherac8dab722025-05-08 15:57:42 +053011#include "NvidiaGpuPowerSensor.hpp"
Harshit Agherafa2a5b92025-05-22 11:35:39 +053012#include "NvidiaGpuSensor.hpp"
13
Harshit Aghera128c91d2025-05-27 14:20:24 +053014#include <NvidiaGpuEnergySensor.hpp>
Harshit Agherafa2a5b92025-05-22 11:35:39 +053015#include <boost/asio/io_context.hpp>
16#include <boost/asio/steady_timer.hpp>
17#include <sdbusplus/asio/connection.hpp>
18#include <sdbusplus/asio/object_server.hpp>
19
20#include <chrono>
21#include <cstdint>
22#include <memory>
23#include <string>
24
25class GpuDevice
26{
27 public:
28 GpuDevice(const SensorConfigs& configs, const std::string& name,
29 const std::string& path,
30 const std::shared_ptr<sdbusplus::asio::connection>& conn,
31 uint8_t eid, boost::asio::io_context& io,
32 mctp::MctpRequester& mctpRequester,
33 sdbusplus::asio::object_server& objectServer);
34
35 const std::string& getPath() const
36 {
37 return path;
38 }
39
40 private:
41 void makeSensors();
42
43 void read();
44
45 uint8_t eid{};
46
47 std::chrono::milliseconds sensorPollMs;
48
49 boost::asio::steady_timer waitTimer;
50
51 mctp::MctpRequester& mctpRequester;
52
53 std::shared_ptr<sdbusplus::asio::connection> conn;
54
55 sdbusplus::asio::object_server& objectServer;
56
57 std::shared_ptr<NvidiaGpuTempSensor> tempSensor;
Harshit Aghera0e1718c2025-05-05 12:26:35 +053058 std::shared_ptr<NvidiaGpuTempSensor> tLimitSensor;
Harshit Agherac8dab722025-05-08 15:57:42 +053059 std::shared_ptr<NvidiaGpuPowerSensor> powerSensor;
Harshit Aghera128c91d2025-05-27 14:20:24 +053060 std::shared_ptr<NvidiaGpuEnergySensor> energySensor;
Harshit Agherafa2a5b92025-05-22 11:35:39 +053061
62 SensorConfigs configs;
63
64 std::string name;
65
66 std::string path;
67};