Skip to content

P
Paddle

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

通知 2298
Star 20931
Fork 5422
P
Paddle

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

未验证 提交 1431305e 编写于 作者: Z zyfncg 提交者： GitHub
浏览文件
操作

fix kernel backend select bug (#41002)

上级 d101334c
显示空白变更内容
内联 并排
Showing with 12 addition and 9 deletion
paddle/phi/api/lib/kernel_dispatch.cc
浏览文件 @ 1431305e
...@@ -25,6 +25,7 @@ namespace experimental { ...@@ -25,6 +25,7 @@ namespace experimental {
namespace detail { namespace detail {
BackendSet GetTensorBackendSet(const phi::TensorBase& t) { BackendSet GetTensorBackendSet(const phi::TensorBase& t) {
if (t.initialized()) {
BackendSet backend_set(phi::TransToPhiBackend(t.place())); BackendSet backend_set(phi::TransToPhiBackend(t.place()));
switch (t.layout()) { switch (t.layout()) {
case DataLayout::MKLDNN: case DataLayout::MKLDNN:
...@@ -35,6 +36,8 @@ BackendSet GetTensorBackendSet(const phi::TensorBase& t) { ...@@ -35,6 +36,8 @@ BackendSet GetTensorBackendSet(const phi::TensorBase& t) {
break; break;
} }
return backend_set; return backend_set;
}
return BackendSet(Backend::UNDEFINED);
} }
std::size_t CountLeadingZeros(uint64_t val) { std::size_t CountLeadingZeros(uint64_t val) {
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或