fix phi xpu kernel tensor transform (#51306)

fcab331d · zhupengyang · GitHub · 87c5f23b · fcab331d
隐藏空白更改
内联并排

Showing with 3 addition and 1 deletion

paddle/fluid/framework/new_executor/interpreter/data_transfer.cc ...fluid/framework/new_executor/interpreter/data_transfer.cc +3 -1

未找到文件。
--- a/paddle/fluid/framework/new_executor/interpreter/data_transfer.cc
+++ b/paddle/fluid/framework/new_executor/interpreter/data_transfer.cc
@@ -585,7 +585,9 @@ void ApplyDataTransform(const OpKernelType& expected_kernel_key,
                     tensor_backend == phi::Backend::XPU) &&
                   !(def_backend == phi::Backend::ONEDNN &&
                     tensor_backend == phi::Backend::CPU)) ||
-                  tensor_in->place().GetType() == AllocationType::GPUPINNED) {
+                  tensor_in->place().GetType() == AllocationType::GPUPINNED ||
+                  (platform::is_xpu_place(expected_kernel_key.place_) &&
+                   def_backend == tensor_backend)) {
                expected_kernel_key_for_argument_def =
                    std::make_unique<phi::KernelKey>(
                        def_backend,