code: def matmul_gather_forward_reducescatter_backward(
      input_,
      weight,
      bias,
      process_group,
      async_grad_reduce_scatter,
      dim,
      overlap,
  ):
      return _MatmulWithGatherForwardReduceScatterBackward.apply(
          input_,
          weight,
          bias,
          process_group,
          async_grad_reduce_scatter,
          dim,
          overlap,
      )
  
imports: import torch
  import torch.distributed
  import torch.nn.functional
label_desc: Use image node_id 10 for calling a global function with example usage: matmul_gather_forward_reducescatter_backward(input_, weight, bias, process_group, async_grad_reduce_scatter, dim, overlap) and returns: _MatmulWithGatherForwardReduceScatterBackward