summaryrefslogtreecommitdiffstats
path: root/src/go/collectors/go.d.plugin/modules/nvidia_smi/init.go
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--src/go/collectors/go.d.plugin/modules/nvidia_smi/init.go22
1 files changed, 22 insertions, 0 deletions
diff --git a/src/go/collectors/go.d.plugin/modules/nvidia_smi/init.go b/src/go/collectors/go.d.plugin/modules/nvidia_smi/init.go
new file mode 100644
index 000000000..d8a815bb4
--- /dev/null
+++ b/src/go/collectors/go.d.plugin/modules/nvidia_smi/init.go
@@ -0,0 +1,22 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package nvidia_smi
+
+import (
+ "fmt"
+ "os"
+ "os/exec"
+)
+
+func (nv *NvidiaSMI) initNvidiaSMIExec() (nvidiaSMI, error) {
+ binPath := nv.BinaryPath
+ if _, err := os.Stat(binPath); os.IsNotExist(err) {
+ path, err := exec.LookPath(nv.binName)
+ if err != nil {
+ return nil, fmt.Errorf("error on lookup '%s': %v", nv.binName, err)
+ }
+ binPath = path
+ }
+
+ return newNvidiaSMIExec(binPath, nv.Config, nv.Logger)
+}