3 import matplotlib.pyplot as plt
7 parser = argparse.ArgumentParser(description="Graph memcpy perf")
8 parser.add_argument("--files", nargs='+', type=str, help="files to graph", default=None)
9 args = parser.parse_args()
11 fig, ax = plt.subplots(nrows=1)
14 plt.xlabel("size in bytes")
15 plt.ylabel("BW in GB/s")
16 plt.title("size vs. bw")
19 for arg in args.files:
24 # size: 11430912, perf: 6.76051GB/s, iter: 5
25 line_split = line.split(",")
26 size.append(float(line_split[0].split(":")[1]))
27 perf.append(float(line_split[1].split(":")[1].split("G")[0]))
29 line, = ax.plot(size, perf, '-', linewidth=0.2, label=arg)