class_docstr: This handle assumes that no need to
  copy data out from a contiguous tensor
code: def wait(self) -> None:
      """ """
      # let the current stream to op
      instrument_w_nvtx(self.allgather_handle.wait)()
      if self.complete:
          return
  
      for _, param in enumerate(self.params):
          assert (
              param.ds_status == ZeroParamStatus.INFLIGHT
          ), f"expected param {param.ds_summary()} to be inflight"
          param.ds_status = ZeroParamStatus.AVAILABLE
  
      self.complete = True
  
imports: import sys
  from typing import List
  import deepspeed
  import torch
  from deepspeed import comm
  from deepspeed.runtime.zero.mics_utils import MiCS_CommGroups, create_mics_comm_groups, scale_tensors
  from deepspeed.runtime.zero.parameter_offload import DeepSpeedZeRoOffload, is_zero_param
  from deepspeed.runtime.zero.partition_parameters import Init, AllGatherCoalescedHandle, ZeroParamStatus
  from deepspeed.runtime.zero.stage3 import DeepSpeedZeroOptimizer_Stage3
  from deepspeed.utils import instrument_w_nvtx, log_dist
  from deepspeed.accelerator import get_accelerator
  from torch import Tensor
  from torch.nn import Parameter
label_desc: Use image node_id 2 for calling the MiCS_AllGatherCoalescedHandle obj's underlying member method code with example usage: obj.wait() without return types