Merge branch 'add_ReLUSquaredActivation' of https://github.com/Oneflo…

…w-Inc/oneflow into add_ReLUSquaredActivation
Oneflow-Inc · Aug 21, 2023 · 8ac041c · 8ac041c
2 parents 9b6cc38 + e0614b6
commit 8ac041c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/oneflow/core/autograd/gradient_funcs/activation.cpp b/oneflow/core/autograd/gradient_funcs/activation.cpp
@@ -162,7 +162,7 @@ class SquareReLU : public OpExprGradFunction<SquareReLUCaptureState> {
 
   Maybe<void> Capture(SquareReLUCaptureState* ctx, const TensorTuple& inputs,
                       const TensorTuple& outputs, const AttrMap& attrs) const override {
-    CHECK_EQ_OR_RETURN(inputs.size(), 1);  // NOLINT(maybe-need-error-msg)
+    CHECK_EQ_OR_RETURN(inputs.size(), 1);   // NOLINT(maybe-need-error-msg)
     CHECK_EQ_OR_RETURN(outputs.size(), 1);  // NOLINT(maybe-need-error-msg)
     ctx->requires_grad = inputs.at(0)->requires_grad();
     if (!ctx->requires_grad) { return Maybe<void>::Ok(); }