Skip to content

Commit 4f67977

Browse files
committed
Fix
1 parent ab65226 commit 4f67977

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

paddle/fluid/operators/pull_gpups_sparse_op.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,8 @@ class PushGpuPSSparseOpMaker : public framework::SingleGradOpMaker<T> {
117117
op->SetType("push_gpups_sparse");
118118
op->SetInput("Ids", this->Input("Ids"));
119119
op->SetInput(framework::GradVarName("Out"), this->OutputGrad("Out"));
120-
op->SetOutput(framework::GradVarName("Ids"), this->InputGrad("Ids", false));
120+
op->SetOutput(framework::GradVarName("Ids"),
121+
this->InputGrad("Ids", /* drop_empty_grad */ false));
121122
op->SetAttrMap(this->Attrs());
122123
}
123124
};

0 commit comments

Comments
 (0)