enable softmax in amp pass (#49189)

0f2683d6 · MarDino · GitHub · 3b90a7f3 · 0f2683d6
隐藏空白更改
内联并排

Showing with 0 addition and 1 deletion

paddle/fluid/framework/ir/auto_mixed_precision_pass.cc paddle/fluid/framework/ir/auto_mixed_precision_pass.cc +0 -1

未找到文件。
--- 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",