def plot_multi_job_stats()

in cluster-trace-gpu-v2020/simulator/utils.py [0:0]

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

def plot_multi_job_stats(npyfiles, to_date=False):
    plt.clf()
    plt.figure(figsize=(12, 6), dpi=120)
    
    for npyfile in npyfiles:
        job_stats = np.load(npyfile)
        job_submit_time, job_duration, job_jct, job_gpu_type, job_num_inst, job_id = job_stats[0], job_stats[1], job_stats[2], job_stats[3], job_stats[4], job_stats[5]
        job_queue_delay = job_jct - job_duration

        try:
            label=ALLOC_POLICY_DICT[int(str(npyfile).split('.log.a')[1].split('-p')[0])]
        except KeyError:
            label = str(npyfile).split('.log.')[1].split('-job_stats.npy')[0]
        plt.plot(job_submit_time, job_queue_delay, alpha=0.5, label=label+'-queue_delay')
        plt.plot(job_submit_time, job_duration, color='grey', alpha=0.3, label='job duration')
    plt.legend(loc='upper left')
    plt.title("Arrival jobs' duration and queueing delay")
    plt.xlabel("Submitted Time")
    plt.ylabel("Run/Wait Time")
    try:
        plt.savefig(str(npyfile).split('.log.')[0]+"-job_stats.png")
    except:
        plt.savefig("job_stats")