code: def fsdp_auto_wrap_bn(module):
      """
      Custom Batch Normalisation FSDP auto wrapper which makes
      sure to use the global group used for all other FSDP wraps
      """
      return fairscale_auto_wrap_bn(
          module, single_rank_pg=False, process_group=get_global_group()
      )
  
file_path: vissl/vissl/utils/fsdp_utils.py
imports: import torch.distributed
  import torch.nn
  from fairscale.nn import auto_wrap, default_auto_wrap_policy, enable_wrap
  from fairscale.nn.data_parallel import auto_wrap_bn, FullyShardedDataParallel
  from vissl.config.attr_dict import AttrDict
label_desc: Use image node_id 3 for calling a global function with example usage: fsdp_auto_wrap_bn(module) and returns: fairscale_auto_wrap_bn