fix data transform problem for cudnn backend (#41622)

c055b50c · zyfncg · GitHub · d6e15914 · c055b50c
显示空白变更内容
内联并排

Showing with 2 addition and 1 deletion

paddle/phi/api/lib/data_transform.cc paddle/phi/api/lib/data_transform.cc +2 -1

未找到文件。
--- a/paddle/phi/api/lib/data_transform.cc
+++ b/paddle/phi/api/lib/data_transform.cc
@@ -40,7 +40,8 @@ inline bool NeedTransformPlace(const paddle::platform::Place& input,
  bool ret =
      input.GetType() == AllocationType::GPUPINNED ||
      (transform_flag.need_trans_backend() && target != Backend::ALL_BACKEND &&
-       phi::TransToPhiBackend(input) != target);
+       phi::TransToPhiBackend(input) !=
+           (target != Backend::GPUDNN ? target : Backend::GPU));
  return ret;
 }