def traced_barrier()

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

• 4 lines of code
• 2 McCabe index (conditional complexity)

def traced_barrier(group=None, async_op=False, device_ids=None):
  """Intercepts invocations of torch.distributed.barrier.
  """
  if _should_rank_record_comm(group):
    _emit_call_description('barrier', message_size=1, group=group)

  return torch.distributed.untraced_barrier(group, async_op, device_ids)