Skip to content

P
Paddle

PaddlePaddle / Paddle
大约 1 年 前同步成功

通知 2298
Star 20931
Fork 5422
P
Paddle
未验证 提交 43b950f7 编写于 作者: Y YuhangLi 提交者： GitHub
浏览文件
操作

[CUSTOM_DEVICE]custom device fetch_v2 support (#53093)

上级 63c83870
显示空白变更内容
内联 并排
Showing with 2 addition and 1 deletion
paddle/fluid/operators/controlflow/fetch_v2_op.cc
浏览文件 @ 43b950f7
...@@ -156,7 +156,8 @@ class FetchV2Kernel { ...@@ -156,7 +156,8 @@ class FetchV2Kernel {
} }
auto *dst_item = &(PADDLE_GET(phi::DenseTensor, fetch_list->at(col))); auto *dst_item = &(PADDLE_GET(phi::DenseTensor, fetch_list->at(col)));
bool check_place = platform::is_cpu_place(src_item.place()) || bool check_place = platform::is_cpu_place(src_item.place()) ||
platform::is_cuda_pinned_place(src_item.place()); platform::is_cuda_pinned_place(src_item.place()) ||
platform::is_custom_place(src_item.place());
PADDLE_ENFORCE_EQ( PADDLE_ENFORCE_EQ(
check_place, check_place,
true, true,
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或