From c46b023d7a165d4b1a32f2b6fdc00d66334597f1 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期六, 11 一月 2020 16:59:55 +0800
Subject: [PATCH] debug
---
app/slave/slave.go | 12 +++++++++++-
1 files changed, 11 insertions(+), 1 deletions(-)
diff --git a/app/slave/slave.go b/app/slave/slave.go
index 8ac53ef..d00d6e2 100644
--- a/app/slave/slave.go
+++ b/app/slave/slave.go
@@ -6,6 +6,7 @@
"analysis/util"
"context"
"plugin"
+ "strconv"
"basic.com/valib/gogpu.git"
)
@@ -49,10 +50,19 @@
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, gpu, shm, util.ToRuleIPC, 30, logo.Infoln, nil)
+ 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
--
Gitblit v1.8.0