Add fused_gemm_epilogue to white list.

Author: Xreki

apps/protein_folding/helixfold/utils/utils.py

@@ -35,6 +35,7 @@ def get_custom_amp_list():
         "fill_any_like",
         "fill_constant",
         "fused_gate_attention",
+        "fused_gemm_epilogue",
         "gather",
         "gaussian_random",
         "layer_norm",
@@ -73,6 +74,7 @@ def get_structure_module_bf16_op_list():
         "fill_any_like",
         "fill_constant",
         "fused_gate_attention",
+        "fused_gemm_epilogue",
         "gaussian_random",
         "linspace",
         "log_softmax",