Skip to content

P
Paddle

马赛克医院 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle

体验新版 GitCode，发现更多精彩内容 >>

提交 71157b33 编写于 作者: Y Yang Yu
浏览文件
操作

Merge branch 'feature/enhance_dev_ctx_pool' into feature/is_nan

上级 837da796 16a84328
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
paddle/pybind/tensor_py.h
浏览文件 @ 71157b33
...@@ -64,7 +64,7 @@ struct CastToPyBufferImpl<true, I, ARGS...> { ...@@ -64,7 +64,7 @@ struct CastToPyBufferImpl<true, I, ARGS...> {
tensor.dims(), platform::CPUPlace())); tensor.dims(), platform::CPUPlace()));
platform::DeviceContextPool &pool = platform::DeviceContextPool &pool =
platform::DeviceContextPool::Instance; platform::DeviceContextPool::Instance();
auto dev_ctx = static_cast<const platform::CUDADeviceContext *>( auto dev_ctx = static_cast<const platform::CUDADeviceContext *>(
pool.Get(tensor.place())); pool.Get(tensor.place()));
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或