Merge branch 'allow-library-selection' into 'main'

Allow path to libnvidia-ml to be specified

See merge request nvidia/cloud-native/go-nvlib!56

pkg/nvml/nvml.go

@@ -30,7 +30,18 @@ type nvmlLib struct {
 var _ Interface = (*nvmlLib)(nil)
 
 // New creates a new instance of the NVML Interface
-func New() Interface {
+func New(opts ...Option) Interface {
+	o := &options{}
+	for _, opt := range opts {
+		opt(o)
+	}
+
+	var nvmlOptions []nvml.LibraryOption
+	if o.libraryPath != "" {
+		nvmlOptions = append(nvmlOptions, nvml.WithLibraryPath(o.libraryPath))
+	}
+	nvml.SetLibraryOptions(nvmlOptions...)
+
 	return &nvmlLib{}
 }
 

pkg/nvml/options.go

@@ -0,0 +1,32 @@
+/**
+# Copyright 2023 NVIDIA CORPORATION
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+**/
+
+package nvml
+
+// options represents the options that could be passed to the nvml contructor.
+type options struct {
+	libraryPath string
+}
+
+// Option represents a functional option to control behaviour.
+type Option func(*options)
+
+// WithLibraryPath sets the NVML library name to use.
+func WithLibraryPath(libraryPath string) Option {
+	return func(o *options) {
+		o.libraryPath = libraryPath
+	}
+}