fix transform data (#44266)

* fix transform data * fix dropout kernel * Revert "fix transform data" This reverts commit ada75ecd169ea194ce43f7ed75dcc968f5ed2fb9.

fix transform data (#44266)
* fix transform data * fix dropout kernel * Revert "fix transform data" This reverts commit ada75ecd169ea194ce43f7ed75dcc968f5ed2fb9.
abc2cc57 · Roc · GitHub · 469d5ab4 · abc2cc57
显示空白变更内容
内联并排

Showing with 6 addition and 2 deletion

paddle/phi/kernels/gpu/dropout_kernel.cu paddle/phi/kernels/gpu/dropout_kernel.cu +6 -2

未找到文件。
--- a/paddle/phi/kernels/gpu/dropout_kernel.cu
+++ b/paddle/phi/kernels/gpu/dropout_kernel.cu
@@ -84,7 +84,9 @@ PD_REGISTER_KERNEL(dropout,
                   float,
                   double,
                   phi::dtype::bfloat16,
-                   phi::dtype::float16) {}
+                   phi::dtype::float16) {
+  kernel->InputAt(1).SetBackend(phi::Backend::ALL_BACKEND);
+}

 PD_REGISTER_KERNEL(dropout_nd,
                   GPU,
@@ -93,4 +95,6 @@ PD_REGISTER_KERNEL(dropout_nd,
                   float,
                   double,
                   phi::dtype::bfloat16,
-                   phi::dtype::float16) {}
+                   phi::dtype::float16) {
+  kernel->InputAt(1).SetBackend(phi::Backend::ALL_BACKEND);
+}