From 0f2683d669323356d9c3595088bda76ba7fe681a Mon Sep 17 00:00:00 2001
From: ZZK <359521840@qq.com>
Date: Fri, 23 Dec 2022 11:15:17 +0800
Subject: [PATCH] enable softmax in amp pass (#49189)

---
 paddle/fluid/framework/ir/auto_mixed_precision_pass.cc | 1 -
 1 file changed, 1 deletion(-)

diff --git a/paddle/fluid/framework/ir/auto_mixed_precision_pass.cc b/paddle/fluid/framework/ir/auto_mixed_precision_pass.cc
index 44b41a89700..1393cd9d091 100644
--- a/paddle/fluid/framework/ir/auto_mixed_precision_pass.cc
+++ b/paddle/fluid/framework/ir/auto_mixed_precision_pass.cc
@@ -178,7 +178,6 @@ void AutoMixedPrecisionPass::SetDefaultBlacklist() const {
       "rsqrt",
       "sum",
       "cos_sim",
-      "softmax",
       "softmax_with_cross_entropy",
       "sigmoid_cross_entropy_with_logits",
       "c_softmax_with_cross_entropy",
-- 
GitLab