def _get_original_p2p_op()

in sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py [0:0]

• 5 lines of code
• 3 McCabe index (conditional complexity)

def _get_original_p2p_op(op):
  if op == torch.distributed.isend:
    return torch.distributed.untraced_isend
  elif op == torch.distributed.irecv:
    return torch.distributed.untraced_irecv