Skip to content

Commit a9aebfa

Browse files
authored
Merge pull request #14758 from JiabinYang/fix_hs_op_release
fix bug in dist train on hs, test=release/1.2
2 parents 4a29292 + 27df187 commit a9aebfa

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

paddle/fluid/operators/hierarchical_sigmoid_op.cc

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -150,13 +150,12 @@ class HierarchicalSigmoidGradOp : public framework::OperatorWithKernel {
150150
"Output(W@Grad should not be null.");
151151
PADDLE_ENFORCE(ctx->HasOutput(framework::GradVarName("X")),
152152
"Output(X@Grad should not be null.");
153-
if (!ctx->Attrs().Get<bool>("is_sparse")) {
154-
if (ctx->HasOutput(framework::GradVarName("Bias"))) {
155-
ctx->SetOutputDim(framework::GradVarName("Bias"),
156-
ctx->GetInputDim("Bias"));
157-
}
158-
ctx->SetOutputDim(framework::GradVarName("W"), ctx->GetInputDim("W"));
153+
154+
if (ctx->HasOutput(framework::GradVarName("Bias"))) {
155+
ctx->SetOutputDim(framework::GradVarName("Bias"),
156+
ctx->GetInputDim("Bias"));
159157
}
158+
ctx->SetOutputDim(framework::GradVarName("W"), ctx->GetInputDim("W"));
160159
ctx->SetOutputDim(framework::GradVarName("X"), ctx->GetInputDim("X"));
161160
}
162161

0 commit comments

Comments
 (0)