def plot_multi_cluster_util()

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

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

def plot_multi_cluster_util(npyfiles, to_date=False):
    plt.clf()
    plt.figure(figsize=(12, 6), dpi=120)
    
    for npyfile in npyfiles:
        cluster_util = np.load(npyfile)
        cluster_time, cluster_cpu, cluster_gpu = cluster_util[0], cluster_util[1], cluster_util[2]

        try:
            label=ALLOC_POLICY_DICT[int(str(npyfile).split('.log.a')[1].split('-p')[0])]
        except KeyError:
            label = str(npyfile).split('.log.')[1].split('-cluster_util.npy')[0]
        plt.plot(cluster_time, cluster_gpu, alpha=0.5, label=label+'-GPU')
    plt.legend(loc='upper left')
    plt.title("Cluster Utilization")
    plt.xlabel("Time")
    plt.ylabel("Resource")
    try:
        plt.savefig(str(npyfile).split('.log.')[0]+"-cluster_util.png")
    except:
        plt.savefig("cluster_util")