Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
机器未来
Paddle
提交
70b67f10
P
Paddle
项目概览
机器未来
/
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看板
未验证
提交
70b67f10
编写于
3月 26, 2021
作者:
W
Wilber
提交者:
GitHub
3月 26, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix go api bug. (#31857)
上级
e804f085
变更
7
隐藏空白更改
内联
并排
Showing
7 changed file
with
14 addition
and
10 deletion
+14
-10
go/README_cn.md
go/README_cn.md
+1
-0
go/demo/mobilenet.go
go/demo/mobilenet.go
+1
-1
go/paddle/common.go
go/paddle/common.go
+1
-1
go/paddle/config.go
go/paddle/config.go
+1
-1
go/paddle/predictor.go
go/paddle/predictor.go
+2
-2
go/paddle/tensor.go
go/paddle/tensor.go
+2
-2
paddle/fluid/inference/capi/pd_predictor.cc
paddle/fluid/inference/capi/pd_predictor.cc
+6
-3
未找到文件。
go/README_cn.md
浏览文件 @
70b67f10
...
...
@@ -50,6 +50,7 @@ output_data := value.Interface().([][]float32)
运行
```
bash
go mod init github.com/paddlepaddle
export
LD_LIBRARY_PATH
=
`
pwd
`
/paddle_c/paddle/lib:
$LD_LIBRARY_PATH
go run ./demo/mobilenet.go
```
go/demo/mobilenet.go
浏览文件 @
70b67f10
...
...
@@ -13,7 +13,7 @@
// limitations under the License.
package
main
import
"
..
/paddle"
import
"
github.com/paddlepaddle
/paddle"
import
"strings"
import
"io/ioutil"
import
"strconv"
...
...
go/paddle/common.go
浏览文件 @
70b67f10
...
...
@@ -15,7 +15,7 @@
package
paddle
// #cgo CFLAGS: -I${SRCDIR}/../paddle_c/paddle/include
// #cgo LDFLAGS: -L${SRCDIR}/../paddle_c/paddle/lib -lpaddle_
fluid
_c
// #cgo LDFLAGS: -L${SRCDIR}/../paddle_c/paddle/lib -lpaddle_
inference
_c
// #include <stdbool.h>
// #include <paddle_c_api.h>
import
"C"
...
...
go/paddle/config.go
浏览文件 @
70b67f10
...
...
@@ -15,7 +15,7 @@
package
paddle
// #cgo CFLAGS: -I${SRCDIR}/../paddle_c/paddle/include
// #cgo LDFLAGS: -L${SRCDIR}/../paddle_c/paddle/lib -lpaddle_
fluid
_c
// #cgo LDFLAGS: -L${SRCDIR}/../paddle_c/paddle/lib -lpaddle_
inference
_c
// #include <stdbool.h>
// #include <stdlib.h>
// #include <paddle_c_api.h>
...
...
go/paddle/predictor.go
浏览文件 @
70b67f10
...
...
@@ -15,7 +15,7 @@
package
paddle
// #cgo CFLAGS: -I${SRCDIR}/../paddle_c/paddle/include
// #cgo LDFLAGS: -L${SRCDIR}/../paddle_c/paddle/lib -lpaddle_
fluid
_c
// #cgo LDFLAGS: -L${SRCDIR}/../paddle_c/paddle/lib -lpaddle_
inference
_c
// #include <stdbool.h>
// #include "paddle_c_api.h"
import
"C"
...
...
@@ -88,7 +88,7 @@ func (predictor *Predictor) GetInputNames() []string {
}
func
(
predictor
*
Predictor
)
GetOutputNames
()
[]
string
{
names
:=
make
([]
string
,
predictor
.
Get
In
putNum
())
names
:=
make
([]
string
,
predictor
.
Get
Out
putNum
())
for
i
:=
0
;
i
<
len
(
names
);
i
++
{
names
[
i
]
=
predictor
.
GetOutputName
(
i
)
}
...
...
go/paddle/tensor.go
浏览文件 @
70b67f10
...
...
@@ -15,7 +15,7 @@
package
paddle
// #cgo CFLAGS: -I${SRCDIR}/../paddle_c/paddle/include
// #cgo LDFLAGS: -L${SRCDIR}/../paddle_c/paddle/lib -lpaddle_
fluid
_c
// #cgo LDFLAGS: -L${SRCDIR}/../paddle_c/paddle/lib -lpaddle_
inference
_c
// #include <stdbool.h>
// #include <stdlib.h>
// #include <string.h>
...
...
@@ -209,7 +209,7 @@ func DecodeTensor(r *bytes.Reader, shape []int32, t reflect.Type, ptr reflect.Va
value
:=
reflect
.
Indirect
(
ptr
)
value
.
Set
(
reflect
.
MakeSlice
(
t
,
int
(
shape
[
0
]),
int
(
shape
[
0
])))
if
len
(
shape
)
==
1
&&
value
.
Len
()
>
0
{
switch
value
.
Index
(
1
)
.
Kind
()
{
switch
value
.
Index
(
0
)
.
Kind
()
{
case
reflect
.
Uint8
,
reflect
.
Int32
,
reflect
.
Int64
,
reflect
.
Float32
:
binary
.
Read
(
r
,
Endian
(),
value
.
Interface
())
return
...
...
paddle/fluid/inference/capi/pd_predictor.cc
浏览文件 @
70b67f10
...
...
@@ -207,13 +207,16 @@ int PD_GetOutputNum(const PD_Predictor* predictor) {
}
const
char
*
PD_GetInputName
(
const
PD_Predictor
*
predictor
,
int
n
)
{
static
std
::
vector
<
std
::
string
>
names
=
predictor
->
predictor
->
GetInputNames
();
static
std
::
vector
<
std
::
string
>
names
;
names
.
resize
(
predictor
->
predictor
->
GetInputNames
().
size
());
names
[
n
]
=
predictor
->
predictor
->
GetInputNames
()[
n
];
return
names
[
n
].
c_str
();
}
const
char
*
PD_GetOutputName
(
const
PD_Predictor
*
predictor
,
int
n
)
{
static
std
::
vector
<
std
::
string
>
names
=
predictor
->
predictor
->
GetOutputNames
();
static
std
::
vector
<
std
::
string
>
names
;
names
.
resize
(
predictor
->
predictor
->
GetOutputNames
().
size
());
names
[
n
]
=
predictor
->
predictor
->
GetOutputNames
()[
n
];
return
names
[
n
].
c_str
();
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录