blob: 9bcf64efe1eda20f43789f91d7ac8e2ed20d988c [file] [log] [blame]
Harshit Aghera4ecdfaa2025-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 Aghera902c6492025-05-08 15:57:42 +053011#include "NvidiaGpuPowerSensor.hpp"
Harshit Aghera4ecdfaa2025-05-22 11:35:39 +053012#include "NvidiaGpuSensor.hpp"
13
14#include <boost/asio/io_context.hpp>
15#include <boost/asio/steady_timer.hpp>
16#include <sdbusplus/asio/connection.hpp>
17#include <sdbusplus/asio/object_server.hpp>
18
19#include <chrono>
20#include <cstdint>
21#include <memory>
22#include <string>
Harshit Aghera5e7decc2025-05-07 16:20:16 +053023#include <vector>
Harshit Aghera4ecdfaa2025-05-22 11:35:39 +053024
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
Harshit Aghera5e7decc2025-05-07 16:20:16 +053045 void processTLimitThresholds(uint8_t rc,
46 const std::vector<int32_t>& thresholds);
47
Harshit Aghera4ecdfaa2025-05-22 11:35:39 +053048 uint8_t eid{};
49
50 std::chrono::milliseconds sensorPollMs;
51
52 boost::asio::steady_timer waitTimer;
53
54 mctp::MctpRequester& mctpRequester;
55
56 std::shared_ptr<sdbusplus::asio::connection> conn;
57
58 sdbusplus::asio::object_server& objectServer;
59
60 std::shared_ptr<NvidiaGpuTempSensor> tempSensor;
Harshit Agheraba138da2025-05-05 12:26:35 +053061 std::shared_ptr<NvidiaGpuTempSensor> tLimitSensor;
Harshit Aghera902c6492025-05-08 15:57:42 +053062 std::shared_ptr<NvidiaGpuPowerSensor> powerSensor;
Harshit Aghera4ecdfaa2025-05-22 11:35:39 +053063
64 SensorConfigs configs;
65
66 std::string name;
67
68 std::string path;
69};