Skip to content

Commit 78acbb8

Browse files
authored
Merge pull request PaddlePaddle#359 from jiweibo/stream_doc
add clone stream doc
2 parents fc6b072 + cbc3fff commit 78acbb8

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

docs/api_reference/python_api_doc/Predictor.md

+5
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,11 @@ paddle.inference.Predictor.run()
3838
# 返回:Predictor - 新的 Predictor
3939
paddle.inference.Predictor.clone()
4040

41+
# 根据该 Predictor,和外部 stream, 克隆一个新的 Predictor,两个 Predictor 之间共享权重,克隆 Predictor 绑定外部 stream
42+
# 参数:paddle.device.cuda.Stream
43+
# 返回:Predictor - 新的 Predictor
44+
paddle.inference.Predictor.clone(paddle.device.cuda.Stream(paddle.CUDAPlace(0), 1))
45+
4146
# 释放中间 Tensor
4247
# 参数:None
4348
# 返回:None

0 commit comments

Comments
 (0)