search

PaddlePaddle / Paddle

PArallel Distributed Deep LEarning: Machine Learning Framework from Industrial Practice （『飞桨』核心框架，深度学习&机器学习高性能单机、分布式训练和跨平台部署）
http://www.paddlepaddle.org/
Apache License 2.0
22.28k stars 5.61k forks source link

set_deviceid before call gpu #69633

Open wanghuancoder opened 14 hours ago

wanghuancoder commented 14 hours ago

PR Category

Execute Infrastructure

PR Types

Bug fixes

Description

动态图每次从Python进C++执行GPU前需要更新device id。 自动生成的eager_op_function.cc这么做了。但手写的接口没有做这个处理。本PR给补上。

Pcard-67164

paddle-bot[bot] commented 14 hours ago

你的PR提交成功，感谢你对开源项目的贡献! 请关注后续CI自动化测试结果，详情请参考Paddle-CI手册。 Your PR has been submitted. Thanks for your contribution! Please wait for the result of CI firstly. See Paddle CI Manual for details.