fix: error initializing dcgm

Signed-off-by: Maryam Tahhan <[email protected]>
sustainable-computing-io · Sep 18, 2024 · c20c23c · c20c23c
1 parent 600235f
commit c20c23c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pkg/config/config.go b/pkg/config/config.go
@@ -118,7 +118,7 @@ func newConfig() *Config {
 		Metrics:                getMetricsConfig(),
 		Redfish:                getRedfishConfig(),
 		Libvirt:                getLibvirtConfig(),
-		DCGMHostEngineEndpoint: getConfig("DCGM_HOST_ENGINE_ENDPOINT", ""),
+		DCGMHostEngineEndpoint: getConfig("NVIDIA_HOSTENGINE_ENDPOINT", defaultDCGMHostEngineEndpoint),
 		KernelVersion:          float32(0),
 	}
 }

diff --git a/pkg/config/types.go b/pkg/config/types.go
@@ -103,4 +103,5 @@ const (
 	defaultProcessPlatformPowerKey     = "PROCESS_TOTAL"
 	defaultProcessComponentsPowerKey   = "PROCESS_COMPONENTS"
 	DefaultMachineSpecFilePath         = "/etc/kepler/models/machine/spec.json"
+	defaultDCGMHostEngineEndpoint      = "localhost:5555"
 )