progress on gpu metrics

This commit is contained in:
Henry Dollman
2024-11-08 16:52:50 -05:00
parent b433437636
commit 8262a9a45b
6 changed files with 313 additions and 7 deletions

View File

@@ -24,6 +24,7 @@ type Agent struct {
sensorsContext context.Context // Sensors context to override sys location
sensorsWhitelist map[string]struct{} // List of sensors to monitor
systemInfo system.Info // Host system info
gpuManager *GPUManager // Manages GPU data
}
func NewAgent() *Agent {
@@ -74,6 +75,13 @@ func (a *Agent) Run(pubKey []byte, addr string) {
a.initializeNetIoStats()
a.dockerManager = newDockerManager()
// initialize GPU manager
if gm, err := NewGPUManager(); err != nil {
slog.Error("GPU manager", "err", err)
} else {
a.gpuManager = gm
}
// if debugging, print stats
if a.debug {
slog.Debug("Stats", "data", a.gatherStats())