Skip to content

P
Paddle

Crayon鑫 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 80d91504 编写于 作者: D dongzhihong
浏览文件
操作

" add elementtype in optimizer"

上级 7f038178
隐藏空白更改
内联 并排
Showing with 2 addition and 1 deletion
go/pserver/optimizer.go
浏览文件 @ 80d91504
...@@ -2,7 +2,7 @@ package pserver ...@@ -2,7 +2,7 @@ package pserver
// #cgo pkg-config: protobuf // #cgo pkg-config: protobuf
// #cgo CFLAGS: -I ../../ // #cgo CFLAGS: -I ../../
// FIXME: ldflags contain "build" path // //FIXME: ldflags contain "build" path
// #cgo LDFLAGS: ../../build/go/pserver/cclient/libpaddle_go_optimizer.a -lstdc++ // #cgo LDFLAGS: ../../build/go/pserver/cclient/libpaddle_go_optimizer.a -lstdc++
// #include "paddle/optimizer/optimizer.h" // #include "paddle/optimizer/optimizer.h"
// #include <stdlib.h> // #include <stdlib.h>
...@@ -37,6 +37,7 @@ func cArrayToSlice(p unsafe.Pointer, len int) []byte { ...@@ -37,6 +37,7 @@ func cArrayToSlice(p unsafe.Pointer, len int) []byte {
func newOptimizer(paramWithConfigs ParameterWithConfig) *optimizer { func newOptimizer(paramWithConfigs ParameterWithConfig) *optimizer {
o := &optimizer{} o := &optimizer{}
o.ElementType = paramWithConfigs.Param.ElementType
p := paramWithConfigs.Param p := paramWithConfigs.Param
c := paramWithConfigs.Config c := paramWithConfigs.Config
var cbuffer unsafe.Pointer var cbuffer unsafe.Pointer
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或