package slave
|
|
import (
|
"analysis/app"
|
"analysis/logo"
|
"analysis/util"
|
"context"
|
"plugin"
|
"strconv"
|
|
"basic.com/valib/gogpu.git"
|
)
|
|
// TwoPluginConflict test
|
func TwoPluginConflict(commSoFile, config string) bool {
|
cfg, err := app.ReadConfig(config)
|
if err != nil {
|
logo.Errorln("Slave Read Config Error: ", err)
|
return false
|
}
|
sdk := LoadSDK(cfg.SoFile)
|
if sdk == nil {
|
}
|
|
plug, err := plugin.Open(commSoFile)
|
if err != nil || plug == nil {
|
logo.Errorln("Slave Open so File: ", commSoFile, " Error: ", err)
|
return false
|
}
|
return false
|
}
|
|
// Run run
|
func Run(ctx context.Context, config, typ, id string, gpu int, shm bool) bool {
|
cfg, err := app.ReadConfig(config)
|
if err != nil {
|
logo.Errorln("Slave Read Config Error: ", err)
|
return false
|
}
|
sdk := LoadSDK(cfg.SoFile)
|
if sdk == nil {
|
return false
|
}
|
|
// 配置文件设置了gpu memory且值至少两位数,判断是否有gpu可用
|
if v, ok := cfg.Param["gpu-memory"]; ok && len(v) > 1 {
|
indices := gogpu.RankGPU()
|
if len(indices) == 0 {
|
logo.Errorln("!!!!!!THERE IS NO VALID GPU")
|
return false
|
}
|
}
|
// 设置gpu index
|
rGPU := gpu
|
if v, ok := cfg.Param["gpu"]; ok {
|
if t, err := strconv.Atoi(v); err != nil {
|
rGPU = -1
|
} else {
|
rGPU = t
|
}
|
}
|
// func Create(config string, typ, id string, gpu int, shm bool, ipc2Rule string, ruleMaxSize int, fn func(...interface{}), reserved map[string]string) interface{}
|
// func Run(ctx context.Context, i interface{}) {
|
|
handle := sdk.fnCreate(config, typ, id, rGPU, shm, util.ToRuleIPC, 30, logo.Infoln, nil)
|
if handle == nil {
|
logo.Errorln("Create SDK: ", typ, " ID: ", id, " Error")
|
return false
|
}
|
|
sdk.fnRun(ctx, handle)
|
|
return true
|
}
|
