package gogpu
|
|
/*
|
#cgo CFLAGS: -I. -w -g
|
#cgo CXXFLAGS: -I. -std=c++11 -w -g
|
#cgo LDFLAGS: -ldl
|
#include <stdlib.h>
|
#include "cgpu.h"
|
*/
|
import "C"
|
import (
|
"encoding/json"
|
"errors"
|
"strconv"
|
"strings"
|
"unsafe"
|
)
|
|
// IdleGPU pass needed gpu memory size test if satisfy
|
func IdleGPU(memSize int) int {
|
return int(C.get_idle_gpu(C.int(memSize)))
|
}
|
|
// GpuUnitInfo gpu card info
|
type GpuUnitInfo struct {
|
GpuUtilization int `json:"utilization"`
|
GpuTemperature int `json:"temperature"`
|
GpuDecoderUtil int `json:"decoder"`
|
GpuEncoderUtil int `json:"encoder"`
|
GpuProcesses int `json:"process"`
|
GpuMemoryTotal int64 `json:"memory"`
|
GpuMemoryFree int64 `json:"free"`
|
GpuMemoryUsed int64 `json:"used"`
|
}
|
|
func (u GpuUnitInfo) String() string {
|
s, _ := json.Marshal(u)
|
return string(s)
|
}
|
|
// GpuInfo gpu info
|
type GpuInfo struct {
|
Count int `json:"count"`
|
Info []GpuUnitInfo `json:"info"`
|
}
|
|
func (g GpuInfo) String() string {
|
s, _ := json.Marshal(g)
|
return string(s)
|
}
|
|
// Info gpu infos
|
func Info() (*GpuInfo, error) {
|
|
p := C.get_gpu_info()
|
if p == nil {
|
return nil, errors.New("can't get gpu infos")
|
}
|
defer C.free(unsafe.Pointer(p))
|
|
str := C.GoString(p)
|
|
infos := strings.Split(str, "|")
|
|
length := len(infos)
|
|
var units []GpuUnitInfo
|
var count int
|
for i := 0; i < length; i++ {
|
if infos[i] == "gpu" {
|
count++
|
|
i++
|
u := GpuUnitInfo{}
|
u.GpuUtilization, _ = strconv.Atoi(infos[i])
|
i++
|
u.GpuTemperature, _ = strconv.Atoi(infos[i])
|
i++
|
u.GpuDecoderUtil, _ = strconv.Atoi(infos[i])
|
i++
|
u.GpuEncoderUtil, _ = strconv.Atoi(infos[i])
|
i++
|
u.GpuProcesses, _ = strconv.Atoi(infos[i])
|
i++
|
u.GpuMemoryTotal, _ = strconv.ParseInt(infos[i], 10, 64)
|
i++
|
u.GpuMemoryFree, _ = strconv.ParseInt(infos[i], 10, 64)
|
i++
|
u.GpuMemoryUsed, _ = strconv.ParseInt(infos[i], 10, 64)
|
units = append(units, u)
|
}
|
}
|
|
return &GpuInfo{count, units}, nil
|
}
|
