update

zhangmeng

2019-07-11 197411370251b04e97a9a60e25b565cdfedbabac

 gogpu.go

@@ -12,6 +12,7 @@
   "encoding/json"
   "errors"
   "fmt"
   "sort"
   "strconv"
   "strings"
   "unsafe"
@@ -22,11 +23,37 @@
   return int(C.get_idle_gpu(C.int(memSize)))
}

// GPURank sort
type GPURank []GpuUnitInfo

func (a GPURank) Len() int           { return len(a) }
func (a GPURank) Swap(i, j int)      { a[i], a[j] = a[j], a[i] }
func (a GPURank) Less(i, j int) bool { return a[i].GpuMemoryFree < a[j].GpuMemoryFree }

// RankGPU rank
func RankGPU() []int {
   info, err := Info()
   if err != nil {
      fmt.Println("RankGPU no gpu")
      return nil
   }
   var ret []int
   if info.Count == 1 {
      ret = append(ret, 0)
   } else {
      sort.Sort(GPURank(info.Info))
      for _, v := range info.Info {
         ret = append(ret, v.GpuIndex)
      }
   }
   return ret
}

// IdleGPU idle most
func IdleGPU() int {
   info, err := Info()
   if err != nil {
      fmt.Println("no gpu")
      fmt.Println("IdleGPU no gpu")
      return -1
   }
   if info.Count == 1 {
@@ -46,6 +73,7 @@

// GpuUnitInfo gpu card info
type GpuUnitInfo struct {
   GpuIndex       int   `json:"index"`
   GpuUtilization int   `json:"utilization"`
   GpuTemperature int   `json:"temperature"`
   GpuDecoderUtil int   `json:"decoder"`
@@ -91,10 +119,11 @@
   var count int
   for i := 0; i < length; i++ {
      if infos[i] == "gpu" {
         u := GpuUnitInfo{}
         u.GpuIndex = count
         count++

         i++
         u := GpuUnitInfo{}
         u.GpuUtilization, _ = strconv.Atoi(infos[i])
         i++
         u.GpuTemperature, _ = strconv.Atoi(infos[i])