MCPcopy Create free account

hub / github.com/Rivendile/Muri / functions

Functions770 in github.com/Rivendile/Muri

↓ 85 callersMethodadd_argument
(self, *args, **kwargs)
cluster_exp/workloads/models/deep_rl/utils/config.py:81
↓ 65 callersMethod__getitem__
(self, item)
cluster_exp/workloads/models/nlp_model.py:97
↓ 59 callersFunctiontensor
(x)
cluster_exp/workloads/models/deep_rl/utils/torch_utils.py:20
↓ 43 callersMethodclose
(self)
cluster_exp/workloads/models/deep_rl/agent/DQN_agent.py:66
↓ 31 callersMethodmean
(self)
cluster_exp/workloads/models/deep_rl/utils/torch_utils.py:200
↓ 26 callersFunctionlayer_init
(layer, w_scale=1.0)
cluster_exp/workloads/models/deep_rl/network/network_utils.py:23
↓ 25 callersMethodsize
(self)
cluster_exp/workloads/models/deep_rl/component/replay.py:142
↓ 22 callersMethodget_time
(self)
cluster_exp/scheduler.py:28
↓ 19 callersFunctionto_np
(t)
cluster_exp/workloads/models/deep_rl/utils/torch_utils.py:32
↓ 18 callersMethodfeed
(self, data)
cluster_exp/workloads/models/deep_rl/component/replay.py:32
↓ 17 callersMethodstate_dict
(self)
cluster_exp/workloads/models/deep_rl/utils/normalizer.py:21
↓ 16 callersMethodalloc_job_res
alloc job resource
cluster_exp/node.py:162
↓ 16 callersMethodmul
(self, coef)
cluster_exp/workloads/models/deep_rl/utils/torch_utils.py:146
↓ 15 callersMethodsample
(self, batch_size=None)
cluster_exp/workloads/models/deep_rl/component/replay.py:92
↓ 14 callersMethodreset
(self)
cluster_exp/workloads/models/deep_rl/component/envs.py:183
↓ 13 callersMethodalloc_job_res
alloc job resource
simulator/node.py:160
↓ 13 callersMethodload_state_dict
(self, _)
cluster_exp/workloads/models/deep_rl/utils/normalizer.py:24
↓ 13 callersFunctiontry_get_job_res
select placement scheme
cluster_exp/run.py:188
↓ 12 callersMethodfind_runnable_job
(self, job_idx)
cluster_exp/jobs.py:356
↓ 12 callersMethodto_jobinfo
(self, tmp_ejob, is_packing=False)
cluster_exp/jobs.py:780
↓ 12 callersFunctiontry_get_job_res
select placement scheme
simulator/run_sim.py:192
↓ 11 callersMethodcheck_free_cpus
(self)
cluster_exp/node.py:113
↓ 11 callersMethodcheck_free_cpus
(self)
simulator/node.py:111
↓ 11 callersMethodempty_infra
(self)
cluster_exp/cluster.py:94
↓ 11 callersMethodempty_infra
(self)
simulator/cluster.py:94
↓ 11 callersMethodstep
(self, actions)
cluster_exp/workloads/models/deep_rl/component/envs.py:186
↓ 10 callersMethodadd
(self, op)
cluster_exp/workloads/models/deep_rl/utils/torch_utils.py:136
↓ 10 callersMethodfind_runnable_job
(self, job_idx)
simulator/jobs.py:364
↓ 10 callersMethodget
(self, s)
cluster_exp/workloads/models/deep_rl/utils/sum_tree.py:63
↓ 9 callersMethod__init__
(self, action_dim, actor_body_fn, critic_body_fn,
cluster_exp/workloads/models/deep_rl/network/network_heads.py:259
↓ 9 callersMethodjob_complete
['even_time', 'job_id', 'num_gpu', 'submit_time', 'start_time', 'end_time', 'executed time', 'real executed time', 'jct', 'duration', 'pendin
simulator/log.py:339
↓ 9 callersMethodrecord_online_return
(self, info, offset=0)
cluster_exp/workloads/models/deep_rl/agent/BaseAgent.py:63
↓ 8 callersMethodadd_network_load
(self, in_load=0, out_load=0)
cluster_exp/node.py:142
↓ 8 callersMethodadd_network_load
(self, in_load=0, out_load=0)
simulator/node.py:140
↓ 8 callersMethodcalc_iteration_time
(self, packing=None, ordering=1, calc_weight=False)
simulator/matching.py:54
↓ 8 callersMethodcheckpoint
Record cluster, and job information, including: time idle_node busy_node: gpu running full_node: all gpus are
simulator/log.py:158
↓ 8 callersMethodcomm
sync for communication
cluster_exp/workloads/models/cv_model.py:95
↓ 8 callersMethoddata_size
(self)
cluster_exp/workloads/models/cv_model.py:104
↓ 8 callersMethoddist
(self, e)
cluster_exp/matching.py:227
↓ 8 callersMethoddist
(self, e)
simulator/matching.py:244
↓ 8 callersMethodforward_backward
forward, calculate loss and backward
cluster_exp/workloads/models/cv_model.py:80
↓ 8 callersMethodget_data
get data
cluster_exp/workloads/models/cv_model.py:64
↓ 8 callersMethodget_free_gpus
(self, priority)
cluster_exp/node.py:57
↓ 8 callersMethodjob_complete
['even_time', 'job_id', 'num_gpu', 'submit_time', 'start_time', 'end_time', 'executed time', 'real executed time', 'jct', 'duration', 'pendin
cluster_exp/log.py:268
↓ 8 callersMethodprepare
prepare dataloader, model, optimizer for training
cluster_exp/workloads/models/cv_model.py:17
↓ 8 callersMethodprint_info
(self)
cluster_exp/workloads/models/cv_model.py:101
↓ 8 callersMethodrelease_job_res
input is node_dict from placement {'id':xx, 'num_gpu':xxx, 'num_cpu': xxx, 'network': xxxx, 'tasks': [w2, ps2]}
simulator/node.py:184
↓ 8 callersMethodwait
(self)
cluster_exp/task.py:81
↓ 7 callersMethodcheck_free_gpus
(self)
cluster_exp/node.py:54
↓ 7 callersMethodcheck_free_gpus
(self)
simulator/node.py:52
↓ 7 callersMethodclone
(self)
cluster_exp/workloads/models/deep_rl/utils/torch_utils.py:159
↓ 7 callersMethodmove_to_runnable
job gets into the system: pending or running, and finally END
simulator/jobs.py:511
↓ 7 callersMethodrelease_job_res
input is node_dict from placement {'id':xx, 'num_gpu':xxx, 'num_cpu': xxx, 'network': xxxx, 'tasks': [w2, ps2]}
cluster_exp/node.py:186
↓ 6 callersMethodcalc_iteration_time
(self, packing=None, ordering=1)
cluster_exp/matching.py:47
↓ 6 callersMethodcreate_single_node_placement
under this switch, there is only one need used {'switch': xx, 'nodes': [{'id':xx, 'num_gpu':xxx, 'num_cpu': xxx, 'network': xxxx, 'ta
cluster_exp/jobs.py:426
↓ 6 callersMethodcreate_single_node_placement
under this switch, there is only one need used {'switch': xx, 'nodes': [{'id':xx, 'num_gpu':xxx, 'num_cpu': xxx, 'network': xxxx, 'ta
simulator/jobs.py:434
↓ 6 callersMethodexecute
(self, job_info)
cluster_exp/controller.py:50
↓ 6 callersMethodget_free_gpus
(self, priority)
simulator/node.py:55
↓ 6 callersMethodhas_ready_jobs
(self, tmp_time)
cluster_exp/scheduler.py:76
↓ 6 callersMethodkill
(self, job_info)
cluster_exp/controller.py:55
↓ 6 callersFunctionmkdir
(path)
cluster_exp/workloads/models/deep_rl/utils/misc.py:46
↓ 6 callersMethodmove_to_runnable
job gets into the system: pending or running, and finally END
cluster_exp/jobs.py:503
↓ 6 callersFunctionmy_cmp
(x,y)
cluster_exp/matching.py:12
↓ 6 callersFunctionmy_cmp
(x,y)
simulator/matching.py:19
↓ 6 callersMethodq_push
(self, x)
simulator/matching.py:260
↓ 6 callersFunctionread_csv
(file:str, result:dict)
cluster_exp/draw.py:11
↓ 6 callersMethodreset_noise
(self)
cluster_exp/workloads/models/deep_rl/network/network_bodies.py:65
↓ 6 callersMethodset_match
(self, u, v)
simulator/matching.py:303
↓ 5 callersFunctionaware
(x:str, ax=None)
cluster_exp/draw.py:37
↓ 5 callersMethoddone
(self, job_id, job_counter, worker_id, gpus, returncode)
cluster_exp/runtime/rpc/worker_client.py:40
↓ 5 callersMethodextract
(self, keys)
cluster_exp/workloads/models/deep_rl/component/replay.py:50
↓ 5 callersMethodplaceholder
(self)
cluster_exp/workloads/models/deep_rl/component/replay.py:38
↓ 5 callersMethodq_push
(self, x)
cluster_exp/matching.py:243
↓ 5 callersMethodregister
(self)
cluster_exp/worker.py:42
↓ 5 callersMethodset_read_only
(self)
cluster_exp/workloads/models/deep_rl/utils/normalizer.py:15
↓ 5 callersFunctionunaware
(x:str, ax=None)
cluster_exp/draw.py:53
↓ 5 callersMethodunset_read_only
(self)
cluster_exp/workloads/models/deep_rl/utils/normalizer.py:18
↓ 4 callersMethod__init__
(self, name, num_envs=1, single_process=True,
cluster_exp/workloads/models/deep_rl/component/envs.py:154
↓ 4 callersMethodcheckpoint_utils
Record cluster, and job information, including: time queue length blocking index gpu util cpu util
cluster_exp/log.py:189
↓ 4 callersMethodclear_src_utils
(self)
cluster_exp/scheduler.py:88
↓ 4 callersMethoddump_all_logs
(self)
cluster_exp/log.py:119
↓ 4 callersMethodentropy
(self)
cluster_exp/workloads/models/deep_rl/utils/torch_utils.py:98
↓ 4 callersMethodfeature
(self, obs)
cluster_exp/workloads/models/deep_rl/network/network_heads.py:162
↓ 4 callersMethodfilter_log_dirs
(self, pattern, negative_pattern=' ', root='./log', **kwargs)
cluster_exp/workloads/models/deep_rl/utils/plot.py:68
↓ 4 callersFunctionget_gittins_index
(a)
simulator/run_sim.py:676
↓ 4 callersFunctionget_model
(idx, args, sargs)
cluster_exp/workloads/main_real_util.py:46
↓ 4 callersFunctionget_model
(idx, args, sargs)
cluster_exp/workloads/main_real_preenv.py:44
↓ 4 callersMethodget_result
(self)
cluster_exp/workloads/main_real_util.py:63
↓ 4 callersMethodhas_running_trainers
(self, running_jobs)
cluster_exp/scheduler.py:82
↓ 4 callersMethodload_log_dirs
(self, dirs, **kwargs)
cluster_exp/workloads/models/deep_rl/utils/plot.py:85
↓ 4 callersFunctionmulti_resource_blossom_same_sim_jobs
new jobs are added to the end of the ending queue but in the queue, shortest (gpu) job first be served and pack other jobs with the same
cluster_exp/run.py:788
↓ 4 callersFunctionmulti_resource_blossom_same_sim_jobs
new jobs are added to the end of the ending queue but in the queue, shortest (gpu) job first be served and pack other jobs with the same
simulator/run_sim.py:933
↓ 4 callersMethodrelease_cpus
release using cpus back to free list
cluster_exp/node.py:128
↓ 4 callersMethodrelease_cpus
release using cpus back to free list
simulator/node.py:126
↓ 4 callersMethodrelease_job_gpu_cpu
input is gpu and cpu
cluster_exp/node.py:203
↓ 4 callersMethodrelease_job_gpu_cpu
input is gpu and cpu
simulator/node.py:201
↓ 4 callersMethodreset_states
(self)
cluster_exp/workloads/models/deep_rl/component/random_process.py:11
↓ 4 callersMethodset_start_time
(self)
cluster_exp/controller.py:28
↓ 4 callersMethodterminate
(self)
cluster_exp/task.py:78
↓ 3 callersMethod__init__
(self, read_only=False)
cluster_exp/workloads/models/deep_rl/utils/normalizer.py:12
next →1–100 of 770, ranked by callers