Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Crayon鑫
Paddle
提交
27fdccc3
P
Paddle
项目概览
Crayon鑫
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
27fdccc3
编写于
5月 19, 2017
作者:
H
Helin Wang
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix according to comments
上级
9920a06c
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
17 addition
and
21 deletion
+17
-21
paddle/go/pserver/service.go
paddle/go/pserver/service.go
+5
-10
paddle/go/pserver/service_test.go
paddle/go/pserver/service_test.go
+12
-11
未找到文件。
paddle/go/pserver/service.go
浏览文件 @
27fdccc3
...
...
@@ -9,8 +9,7 @@ import (
// ElementType is the type of elements of a Parameter.
type
ElementType
int
var
ErrUnintialized
=
errors
.
New
(
"pserver not initialized"
)
var
ErrAlreadyIntialized
=
errors
.
New
(
"pserver already initialized"
)
var
ErrAlreadyInitialized
=
errors
.
New
(
"pserver already initialized"
)
// Supported element types
const
(
...
...
@@ -56,7 +55,7 @@ func NewService() *Service {
func
(
s
*
Service
)
BeginInitParams
(
config
[]
byte
,
dummy
*
int
)
error
{
select
{
case
<-
s
.
initialized
:
return
ErrAlreadyIntialized
return
ErrAlreadyIn
i
tialized
default
:
}
...
...
@@ -75,7 +74,7 @@ func (s *Service) BeginInitParams(config []byte, dummy *int) error {
func
(
s
*
Service
)
InitParam
(
paramWithConfigs
ParameterWithConfig
,
dummy
*
int
)
error
{
select
{
case
<-
s
.
initialized
:
return
ErrAlreadyIntialized
return
ErrAlreadyIn
i
tialized
default
:
}
...
...
@@ -94,7 +93,7 @@ func (s *Service) InitParam(paramWithConfigs ParameterWithConfig, dummy *int) er
func
(
s
*
Service
)
FinishInitParams
(
dummy0
int
,
dummy1
*
int
)
error
{
select
{
case
<-
s
.
initialized
:
return
ErrAlreadyIntialized
return
ErrAlreadyIn
i
tialized
default
:
}
...
...
@@ -103,11 +102,7 @@ func (s *Service) FinishInitParams(dummy0 int, dummy1 *int) error {
}
func
(
s
*
Service
)
SendGrads
(
grads
[]
Gradient
,
dummy
*
int
)
error
{
select
{
case
<-
s
.
initialized
:
default
:
return
ErrUnintialized
}
<-
s
.
initialized
count
:=
len
(
grads
)
if
count
==
0
{
...
...
paddle/go/pserver/service_test.go
浏览文件 @
27fdccc3
...
...
@@ -98,21 +98,12 @@ func TestMultipleInit(t *testing.T) {
}
err
=
s
.
FinishInitParams
(
0
,
&
dummy
)
if
err
!=
pserver
.
ErrAlreadyIntialized
{
if
err
!=
pserver
.
ErrAlreadyIn
i
tialized
{
t
.
FailNow
()
}
err
=
s
.
BeginInitParams
(
nil
,
&
dummy
)
if
err
!=
pserver
.
ErrAlreadyIntialized
{
t
.
FailNow
()
}
}
func
TestUninitialized
(
t
*
testing
.
T
)
{
s
:=
pserver
.
NewService
()
var
dummy
int
err
:=
s
.
SendGrads
(
nil
,
&
dummy
)
if
err
!=
pserver
.
ErrUnintialized
{
if
err
!=
pserver
.
ErrAlreadyInitialized
{
t
.
FailNow
()
}
}
...
...
@@ -140,6 +131,16 @@ func TestBlockUntilInitialized(t *testing.T) {
wg
.
Done
()
}()
wg
.
Add
(
1
)
go
func
()
{
var
dummy
int
err
:=
s
.
SendGrads
(
nil
,
&
dummy
)
if
err
!=
nil
{
t
.
FailNow
()
}
wg
.
Done
()
}()
var
dummy
int
err
:=
s
.
BeginInitParams
(
nil
,
&
dummy
)
if
err
!=
nil
{
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录