ClearML/go-nvlib
1
0
Fork 0
mirror of https://github.com/clearml/go-nvlib synced 2025-06-26 18:28:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update go-nvml dependency

Browse Source 
This updates go-nvml to include fixes for getting devices by UUID
and to remove the panic from calls to Init and Shutdown.

Signed-off-by: Evan Lezar <elezar@nvidia.com>
This commit is contained in:
Evan Lezar
2022-09-01 14:06:25 +02:00
parent 9175bde20b
commit da71bc2bff
5 changed files with 9 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
vendor/github.com/NVIDIA/go-nvml/pkg/nvml/init.go generated vendored
View File

@@ -32,13 +32,9 @@ var nvml *dl.DynamicLibrary
// nvml.Init()
func Init() Return {
lib := dl.New(nvmlLibraryName, nvmlLibraryLoadFlags)
if lib == nil {
panic(fmt.Sprintf("error instantiating DynamicLibrary for %s", nvmlLibraryName))
}
err := lib.Open()
if err != nil {
panic(fmt.Sprintf("error opening %s: %v", nvmlLibraryName, err))
return ERROR_LIBRARY_NOT_FOUND
}
nvml = lib
@@ -50,13 +46,9 @@ func Init() Return {
// nvml.InitWithFlags()
func InitWithFlags(Flags uint32) Return {
lib := dl.New(nvmlLibraryName, nvmlLibraryLoadFlags)
if lib == nil {
panic(fmt.Sprintf("error instantiating DynamicLibrary for %s", nvmlLibraryName))
}
err := lib.Open()
if err != nil {
panic(fmt.Sprintf("error opening %s: %v", nvmlLibraryName, err))
return ERROR_LIBRARY_NOT_FOUND
}
nvml = lib