diff --git a/python/paddle/amp/amp_lists.py b/python/paddle/amp/amp_lists.py index 645370a28facf6349d38c1e93c8c5b58bcf3a3b0..3aefd298340a533d325d403131579c015752814f 100644 --- a/python/paddle/amp/amp_lists.py +++ b/python/paddle/amp/amp_lists.py @@ -23,6 +23,9 @@ FP16_WHITE_LIST = { 'mul', 'fake_quantize_dequantize_abs_max', 'fake_quantize_dequantize_moving_average_abs_max', + 'fused_gemm_epilogue', + 'fused_attention', + 'fused_feedforward', } # The set of ops that support fp16 calculation and are considered numerically-