diff options
Diffstat (limited to '')
-rw-r--r-- | src/go/collectors/go.d.plugin/modules/nvidia_smi/init.go | 22 |
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) +} |