add gpu usage

gpu.go

@@ -11,18 +11,18 @@ import (
 func getGPU() ([]DeviceInfo, error) {
 	ret := nvml.Init()
 	if ret != nvml.SUCCESS {
-		return nil, fmt.Errorf(ret)
+		return nil, fmt.Errorf(nvml.ErrorString(ret))
 	}
 	defer func() {
 		ret := nvml.Shutdown()
 		if ret != nvml.SUCCESS {
-			return nil, fmt.Errorf(err)
+			return nil, fmt.Errorf(nvml.ErrorString(ret))
 		}
 	}()
 
 	count, ret := nvml.DeviceGetCount()
 	if ret != nvml.SUCCESS {
-		return nil, fmt.Errorf(err)
+		return nil, fmt.Errorf(nvml.ErrorString(ret))
 	}
 
 	res := make([]DeviceInfo, 0, count)
@@ -31,7 +31,7 @@ func getGPU() ([]DeviceInfo, error) {
 
 		device, ret := nvml.DeviceGetHandleByIndex(i)
 		if ret != nvml.SUCCESS {
-			return nil, fmt.Errorf(err)
+			return nil, fmt.Errorf(nvml.ErrorString(ret))
 		}
 
 		// memory, err := device.GetMemoryInfo()
@@ -44,14 +44,14 @@ func getGPU() ([]DeviceInfo, error) {
 		// 	//panic(err)
 		// }
 
-		name, err := device.GetName()
-		if err != nvml.SUCCESS {
-			return nil, fmt.Errorf(err)
+		name, ret := device.GetName()
+		if ret != nvml.SUCCESS {
+			return nil, fmt.Errorf(nvml.ErrorString(ret))
 		}
 
-		memory, err := device.GetMemoryInfo()
-		if err != nvml.SUCCESS {
-			return nil, fmt.Errorf(err)
+		memory, ret := device.GetMemoryInfo()
+		if ret != nvml.SUCCESS {
+			return nil, fmt.Errorf(nvml.ErrorString(ret))
 			//fmt.Println(err)
 		}
 
@@ -99,19 +99,19 @@ func getGpuUsage() ([]DeviceInfo, error) {
 
 	ret := nvml.Init()
 	if ret != nvml.SUCCESS {
-		return nil, fmt.Errorf(ret)
+		return nil, fmt.Errorf(nvml.ErrorString(ret))
 	}
 	defer func() {
 		ret := nvml.Shutdown()
 		if ret != nvml.SUCCESS {
-			return nil, fmt.Errorf(ret)
+			return nil, fmt.Errorf(nvml.ErrorString(ret))
 			//log.Fatalf("Unable to shutdown NVML: %v", nvml.ErrorString(ret))
 		}
 	}()
 
 	count, ret := nvml.DeviceGetCount()
 	if ret != nvml.SUCCESS {
-		return nil, fmt.Errorf(ret)
+		return nil, fmt.Errorf(nvml.ErrorString(ret))
 		//log.Fatalf("Unable to get device count: %v", nvml.ErrorString(ret))
 	}
 
@@ -121,20 +121,20 @@ func getGpuUsage() ([]DeviceInfo, error) {
 
 		device, ret := nvml.DeviceGetHandleByIndex(i)
 		if ret != nvml.SUCCESS {
-			return nil, fmt.Errorf(ret)
+			return nil, fmt.Errorf(nvml.ErrorString(ret))
 			//log.Fatalf("Unable to get device at index %d: %v", i, nvml.ErrorString(ret))
 		}
 
 		name, ret := device.GetName()
 		if ret != nvml.SUCCESS {
-			return nil, fmt.Errorf(ret)
+			return nil, fmt.Errorf(nvml.ErrorString(ret))
 			//panic(err)
 		}
 
 		utilization, ret := device.GetUtilizationRates()
 
 		if ret != nvml.SUCCESS {
-			return nil, fmt.Errorf(ret)
+			return nil, fmt.Errorf(nvml.ErrorString(ret))
 		}
 
 		e := DeviceInfo{