wip: do not use _v2 functions

hashicorp · Aug 20, 2024 · 6c243c3 · 6c243c3
1 parent 8328766
commit 6c243c3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nvml/driver_linux.go b/nvml/driver_linux.go
@@ -110,7 +110,7 @@ func (n *nvmlDriver) DeviceInfoByUUID(uuid string) (*DeviceInfo, error) {
 		return nil, decode("failed to get device name", code)
 	}
 
-	memory, code := nvml.DeviceGetMemoryInfo_v2(device)
+	memory, code := nvml.DeviceGetMemoryInfo(device)
 	if code != nvml.SUCCESS {
 		return nil, decode("failed to get device memory info", code)
 	}
@@ -238,7 +238,7 @@ func (n *nvmlDriver) DeviceInfoAndStatusByUUID(uuid string) (*DeviceInfo, *Devic
 		return nil, nil, decode("failed to get device info", code)
 	}
 
-	mem, code := nvml.DeviceGetMemoryInfo_v2(device)
+	mem, code := nvml.DeviceGetMemoryInfo(device)
 	if code != nvml.SUCCESS {
 		return nil, nil, decode("failed to get device memory utilization", code)
 	}