add protection

zhangmeng

2019-07-11 fb217f9ffc1b40d49feae8d4683efa5d8c5da1c9

add protection

 gogpu.go

@@ -23,6 +23,21 @@
    return int(C.get_idle_gpu(C.int(memSize)))
}

// SatisfyGPU satisfy unit "M"
func SatisfyGPU(index, memSize, reserve int) bool {
    info, err := Info()
    if err != nil || info.Count >= index {
        fmt.Println("SatisfyGPU no gpu or index ilegal: ", index, " gpu count : ", info.Count)
        return false
    }

    var M1 int64 = 1024 * 1024
    if info.Info[index].GpuMemoryFree-int64(memSize)*M1 > int64(reserve)*M1 {
        return true
    }
    return false
}

// GPURank sort
type GPURank []GpuUnitInfo