From 970e1f3e1f1ab0b6b4b7c24d6ad4390555a11769 Mon Sep 17 00:00:00 2001
From: lzy <569782149@qq.com>
Date: Wed, 11 Oct 2023 09:43:05 +0800
Subject: [PATCH] fix softmax backward (#57858)

---
 paddle/phi/kernels/gpudnn/softmax_gpudnn.h | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/paddle/phi/kernels/gpudnn/softmax_gpudnn.h b/paddle/phi/kernels/gpudnn/softmax_gpudnn.h
index 9f8dd99b20d6a7..51d9fadc21c1c4 100644
--- a/paddle/phi/kernels/gpudnn/softmax_gpudnn.h
+++ b/paddle/phi/kernels/gpudnn/softmax_gpudnn.h
@@ -1230,8 +1230,7 @@ bool UseCudnnSoftmax(const GPUContext& ctx,
   }
   constexpr int max_dim = 1024;
   if (!cudnn_available || !last_dim ||
-      (softmax_dim <= max_dim && sizeof(T) <= 4) ||
-      softmax_dim >= MATRIX_SOFTMAX_THREAHOLD) {
+      (softmax_dim <= max_dim && sizeof(T) <= 4)) {
     return false;
   } else {
     return true;