id: 1 unit: def _shunt_torch_communication_calls() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 27 LOC McCabe index: 2 number of parameters: 0 id: 2 unit: def _shunt_torch_communication_calls() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 27 LOC McCabe index: 2 number of parameters: 0 id: 3 unit: def setup() file: sample_workloads/lit-gpt-demo/openwebtext.py start line: 0 end line: 0 size: 22 LOC McCabe index: 1 number of parameters: 2 id: 4 unit: def traced_all_to_all_single() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 19 LOC McCabe index: 3 number of parameters: 6 id: 5 unit: def traced_all_to_all_single() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 19 LOC McCabe index: 3 number of parameters: 6 id: 6 unit: def __init__() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 18 LOC McCabe index: 5 number of parameters: 7 id: 7 unit: def shunt_torch_communication() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 14 LOC McCabe index: 5 number of parameters: 0 id: 8 unit: def _is_crossnode_comm() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 13 LOC McCabe index: 4 number of parameters: 2 id: 9 unit: def _is_crossnode_comm() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 13 LOC McCabe index: 4 number of parameters: 2 id: 10 unit: def __init__() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 13 LOC McCabe index: 4 number of parameters: 6 id: 11 unit: def validate() file: sample_workloads/lit-gpt-demo/openwebtext.py start line: 0 end line: 0 size: 12 LOC McCabe index: 2 number of parameters: 4 id: 12 unit: def validate_args() file: sample_workloads/lit-gpt-demo/openwebtext.py start line: 0 end line: 0 size: 11 LOC McCabe index: 7 number of parameters: 3 id: 13 unit: def _identify_trace_mode() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 11 LOC McCabe index: 1 number of parameters: 0 id: 14 unit: def traced_broadcast_object_list() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 11 LOC McCabe index: 3 number of parameters: 4 id: 15 unit: def traced_reduce_scatter() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 11 LOC McCabe index: 2 number of parameters: 5 id: 16 unit: def traced_reduce_scatter_tensor() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 11 LOC McCabe index: 2 number of parameters: 5 id: 17 unit: def traced_broadcast_object_list() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 11 LOC McCabe index: 3 number of parameters: 4 id: 18 unit: def traced_reduce_scatter() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 11 LOC McCabe index: 2 number of parameters: 5 id: 19 unit: def traced_reduce_scatter_tensor() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 11 LOC McCabe index: 2 number of parameters: 5 id: 20 unit: def traced_all_to_all() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 10 LOC McCabe index: 3 number of parameters: 4 id: 21 unit: def traced_batch_isend_irecv() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 10 LOC McCabe index: 5 number of parameters: 1 id: 22 unit: def _should_rank_print() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 10 LOC McCabe index: 5 number of parameters: 3 id: 23 unit: def traced_all_to_all() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 10 LOC McCabe index: 3 number of parameters: 4 id: 24 unit: def run() file: sample_workloads/pingpong/pingpong.py start line: 0 end line: 0 size: 10 LOC McCabe index: 3 number of parameters: 1 id: 25 unit: def train() file: sample_workloads/lit-gpt-demo/openwebtext.py start line: 0 end line: 0 size: 9 LOC McCabe index: 1 number of parameters: 0 id: 26 unit: def get_lr() file: sample_workloads/lit-gpt-demo/openwebtext.py start line: 0 end line: 0 size: 9 LOC McCabe index: 5 number of parameters: 5 id: 27 unit: def traced_gather() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 9 LOC McCabe index: 2 number of parameters: 5 id: 28 unit: def traced_scatter() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 9 LOC McCabe index: 2 number of parameters: 5 id: 29 unit: def _should_rank_record_comm() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 9 LOC McCabe index: 6 number of parameters: 4 id: 30 unit: def traced_gather() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 9 LOC McCabe index: 2 number of parameters: 5 id: 31 unit: def traced_scatter() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 9 LOC McCabe index: 2 number of parameters: 5 id: 32 unit: def _should_rank_record_comm() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 9 LOC McCabe index: 6 number of parameters: 4 id: 33 unit: def main() file: sample_workloads/lit-gpt-demo/openwebtext.py start line: 0 end line: 0 size: 8 LOC McCabe index: 1 number of parameters: 0 id: 34 unit: def on_train_batch_start() file: sample_workloads/lit-gpt-demo/utilities/nsight_callbacks.py start line: 0 end line: 0 size: 8 LOC McCabe index: 3 number of parameters: 3 id: 35 unit: def shunt_torch_communication() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 8 LOC McCabe index: 2 number of parameters: 0 id: 36 unit: def _should_rank_print() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 8 LOC McCabe index: 4 number of parameters: 3 id: 37 unit: def __iter__() file: sample_workloads/lit-gpt-demo/openwebtext.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 1 id: 38 unit: def traced_all_reduce() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 4 id: 39 unit: def traced_all_gather() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 4 id: 40 unit: def traced_all_gather_into_tensor() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 4 id: 41 unit: def _emit_call_description() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 6 id: 42 unit: def traced_all_reduce() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 4 id: 43 unit: def traced_all_gather() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 4 id: 44 unit: def traced_all_gather_into_tensor() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 4 id: 45 unit: def traced_batch_isend_irecv() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 7 LOC McCabe index: 4 number of parameters: 1 id: 46 unit: def _emit_call_description() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 7 LOC McCabe index: 2 number of parameters: 5 id: 47 unit: def traced_broadcast() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 4 id: 48 unit: def traced_reduce() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 5 id: 49 unit: def traced_broadcast() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 4 id: 50 unit: def traced_reduce() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 6 LOC McCabe index: 2 number of parameters: 5 id: 51 unit: def traced_isend() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 4 id: 52 unit: def traced_irecv() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 4 id: 53 unit: def traced_send() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 4 id: 54 unit: def traced_recv() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 4 id: 55 unit: def _get_original_p2p_op() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 5 LOC McCabe index: 3 number of parameters: 1 id: 56 unit: def traced_isend() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 4 id: 57 unit: def traced_irecv() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 4 id: 58 unit: def traced_send() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 4 id: 59 unit: def traced_recv() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 5 LOC McCabe index: 2 number of parameters: 4 id: 60 unit: def _get_original_p2p_op() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 5 LOC McCabe index: 3 number of parameters: 1 id: 61 unit: def load_datasets() file: sample_workloads/lit-gpt-demo/openwebtext.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 2 id: 62 unit: def __init__() file: sample_workloads/lit-gpt-demo/openwebtext.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 3 id: 63 unit: def _shunt_torch_communication_objects() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 0 id: 64 unit: def traced_barrier() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 3 id: 65 unit: def __init__() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 6 id: 66 unit: def __new__() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 6 id: 67 unit: def _shunt_torch_communication_objects() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 0 id: 68 unit: def traced_barrier() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 4 LOC McCabe index: 2 number of parameters: 3 id: 69 unit: def __init__() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 6 id: 70 unit: def __new__() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 4 LOC McCabe index: 1 number of parameters: 6 id: 71 unit: def __init__() file: sample_workloads/lit-gpt-demo/utilities/nsight_callbacks.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 72 unit: def on_after_backward() file: sample_workloads/lit-gpt-demo/utilities/nsight_callbacks.py start line: 0 end line: 0 size: 3 LOC McCabe index: 2 number of parameters: 1 id: 73 unit: def on_train_epoch_start() file: sample_workloads/lit-gpt-demo/utilities/nsight_callbacks.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 74 unit: def _get_call_source_line() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 75 unit: def _get_call_source_line() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 1 id: 76 unit: def init_processes() file: sample_workloads/pingpong/pingpong.py start line: 0 end line: 0 size: 3 LOC McCabe index: 1 number of parameters: 2 id: 77 unit: def on_train_batch_end() file: sample_workloads/lit-gpt-demo/utilities/nsight_callbacks.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 0 id: 78 unit: def on_before_backward() file: sample_workloads/lit-gpt-demo/utilities/nsight_callbacks.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 79 unit: def _is_current_process_in_group() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 80 unit: def to_json() file: sample_workloads/lit-gpt-demo/utilities/monitor_collectives.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 81 unit: def _is_current_process_in_group() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1 id: 82 unit: def to_json() file: sample_workloads/megatron-gke/docker/monitor_collectives.py start line: 0 end line: 0 size: 2 LOC McCabe index: 1 number of parameters: 1