From 1a4a85920ff1d90bdcd5d2a89a254465211e7e13 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期四, 12 十二月 2019 10:21:40 +0800
Subject: [PATCH] udpate
---
main.go | 15 +++++++++------
1 files changed, 9 insertions(+), 6 deletions(-)
diff --git a/main.go b/main.go
index 15bed52..0c7c0fa 100644
--- a/main.go
+++ b/main.go
@@ -146,18 +146,21 @@
logo.Infoln(os.Args)
ctx, cancel := context.WithCancel(context.Background())
+ ret := false
if role == single {
- proc.SingleRole(ctx, runType, id, gpu, shm)
+ ret = proc.SingleRole(ctx, runType, id, gpu, shm)
} else if role == master {
setParamters()
- proc.MasterRole(ctx)
+ ret = proc.MasterRole(ctx)
} else if role == slave {
- proc.SingleRole(ctx, runType, id, gpu, shm)
+ ret = proc.SingleRole(ctx, runType, id, gpu, shm)
}
- c := make(chan os.Signal, 1)
- signal.Notify(c, os.Interrupt, os.Kill, unix.SIGTERM)
- <-c
+ if ret {
+ c := make(chan os.Signal, 1)
+ signal.Notify(c, os.Interrupt, os.Kill, unix.SIGTERM)
+ <-c
+ }
cancel()
--
Gitblit v1.8.0