code: def linear_with_async_comm(
      input_, weight, bias, parallel_mode, async_grad_allreduce
  ):
      return LinearWithAsyncCommunication.apply(
          input_, weight, bias, parallel_mode, async_grad_allreduce
      )
  
file_path: colossalai/colossalai/legacy/nn/layer/parallel_1d/_operation.py
imports: import torch
  import torch.distributed
  from colossalai.legacy.core import global_context
label_desc: Use image node_id 1 for calling a global function with example usage: linear_with_async_comm(input_, weight, bias, parallel_mode, async_grad_allreduce) and returns: LinearWithAsyncCommunication