code: def get_layer_norm(
      hidden_size,
      eps=1e-5,
      persist_layer_norm=False,
      sequence_parallel=False,
  ):
      # List of hiddens sizes supported in the persistent layer norm kernel
      # If the hidden size is not supported, fall back to the non-persistent
      # kernel.
      persist_ln_hidden_sizes = [
          1024,
          1536,
          2048,
          2304,
          3072,
          3840,
          4096,
          5120,
          6144,
          8192,
          10240,
          12288,
          12800,
          15360,
          16384,
          18432,
          20480,
          24576,
          25600,
          30720,
          32768,
          40960,
          49152,
          65536,
      ]
      if hidden_size not in persist_ln_hidden_sizes:
          persist_layer_norm = False
  
      if persist_layer_norm:
          return FastLayerNorm(
              hidden_size,
              eps,
              sequence_parallel_enabled=sequence_parallel,
          )
      else:
          return MixedFusedLayerNorm(
              hidden_size,
              eps,
              sequence_parallel_enabled=sequence_parallel,
          )
  
label_desc: Use image node_id 1 for calling a global function with example usage: get_layer_norm(hidden_size, eps, persist_layer_norm, sequence_parallel) and returns: FastLayerNorm, MixedFusedLayerNorm