Add output defs for sgd kernel

PaddlePaddle · Mar 7, 2023 · 26fdcde · 26fdcde
1 parent b76c2dc
commit 26fdcde
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/paddle/fluid/framework/new_executor/interpreter/interpreter_util.cc b/paddle/fluid/framework/new_executor/interpreter/interpreter_util.cc
@@ -115,7 +115,6 @@ static std::set<std::string> OpsNeedSetOutputDtypeWhenRegisterPhiKernel = {
     "select",
     "send_recv",
     "send_ue_recv",
-    "sgd",
     "svd",
     "sync_batch_norm_grad",
     "unique",

diff --git a/paddle/phi/kernels/gpu/sgd_kernel.cu b/paddle/phi/kernels/gpu/sgd_kernel.cu
@@ -187,7 +187,9 @@ PD_REGISTER_KERNEL(sgd,
                    phi::SGDDenseKernel,
                    phi::dtype::float16,
                    float,
-                   double) {}
+                   double) {
+  kernel->OutputAt(1).SetDataType(phi::DataType::UNDEFINED);
+}
 
 PD_REGISTER_KERNEL(sgd_dense_param_sparse_grad,
                    GPU,