fix functor

PaddlePaddle · Jan 17, 2022 · 418445b · 418445b
1 parent 5d9c1aa
commit 418445b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/fluid/operators/softmax_cudnn_op.cu.h b/paddle/fluid/operators/softmax_cudnn_op.cu.h
@@ -206,7 +206,7 @@ struct SoftmaxBackwardFunctor {
   HOSTDEVICE inline SoftmaxBackwardFunctor(Tx sum) : sum(sum) {}
 
   HOSTDEVICE inline Ty operator()(const Tx& grad_out, const Tx& out) const {
-    return static_cast<Ty>(grad_out - out * sum);
+    return static_cast<Ty>(out * (grad_out - sum));
   }
 
  private: