Fix bug with wrong instance of lib being called for load/close

Signed-off-by: Kevin Klues <[email protected]>
NVIDIA · Apr 22, 2024 · 8a7ac4d · 8a7ac4d
1 parent a97d07c
commit 8a7ac4d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pkg/nvml/init.go b/pkg/nvml/init.go
@@ -18,15 +18,15 @@ import "C"
 
 // nvml.Init()
 func (l *library) Init() Return {
-	if err := libnvml.load(); err != nil {
+	if err := l.load(); err != nil {
 		return ERROR_LIBRARY_NOT_FOUND
 	}
 	return nvmlInit()
 }
 
 // nvml.InitWithFlags()
 func (l *library) InitWithFlags(flags uint32) Return {
-	if err := libnvml.load(); err != nil {
+	if err := l.load(); err != nil {
 		return ERROR_LIBRARY_NOT_FOUND
 	}
 	return nvmlInitWithFlags(flags)
@@ -39,7 +39,7 @@ func (l *library) Shutdown() Return {
 		return ret
 	}
 
-	err := libnvml.close()
+	err := l.close()
 	if err != nil {
 		return ERROR_UNKNOWN
 	}