def METHOD_NAME():
    if get_env("USE_FAST_LN"):
        paddle.nn.LayerNorm = FastLayerNorm
    elif get_env("USE_FUSED_LN"):
        paddle.nn.LayerNorm = FusedLayerNorm
    elif get_env("USE_FUSED_RMS_NORM"):
        paddle.nn.LayerNorm = FusedRMSNorm
    if get_env("USE_LINEAR_WITH_GRAD_ADD"):
        paddle.nn.functional.linear = FusedLinearWithGradAdd.apply
        paddle.incubate.nn.functional.fused_linear = FusedLinearWithGradAdd.apply