ComputeLibrary/tempschedv2.py at master · adityagupta1089/ComputeLibrary · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
import os
import time
import subprocess
import matplotlib
import matplotlib.pyplot as plt
import numpy as np

from matplotlib.lines import Line2D
from os import environ

pp = "./build/release/examples/"

graph = "graph_resnet50"

cp = ["--target=NEON", "--threads=4"]
gp = ["--target=CL"]
sm = ["taskset", "-c", "0-3"]
bg = ["taskset", "-c", "4-7"]

targets = {
    "cpu_big": lambda g: bg + [pp + g] + cp,
    "cpu_small": lambda g: sm + [pp + g] + cp,
    "gpu": lambda g: [pp + g] + gp
}

T = 5
dt = 0.1
sdt = 0.5
TL = 65000
N = 50

def get_temp():
    with open('/sys/class/thermal/thermal_zone0/temp') as f:
        temp = int(f.readline())
        return temp

def plot_temp(xs, ys, xs2, ys2):
    fig = plt.figure()
    # Plot Lines
    for x, y in zip(xs, ys):
        plt.plot(x, y, color='g')
    for x2, y2 in zip(xs2, ys2):
        plt.plot(x2, y2, color='b')
    # Plot Threshold
    plt.axhline(TL, linestyle='--', color='r')
    # Plot Labels
    plt.xlabel('Time')
    plt.ylabel('Temperature')
    plt.title('Execution ' + graph)
    # Plot Legend
    custom_lines = [
        Line2D([0], [0], color='r', linestyle='--'),
        Line2D([0], [0], color='g'),
        Line2D([0], [0], color='b')]
    plt.legend(custom_lines, ['Threshold', 'Executing', 'Sleeping'])
    # Show Plot
    plt.savefig('tempsched_plots/schedv2.png')

def run_sched():
    env = dict(os.environ)
    env['LD_LIBRARY_PATH'] = './build/release'
    cmds = list(targets[target](graph) for target in targets)
    xs = []
    ys = []
    xs2 = []
    ys2 = []
    t=0
    n=0
    # sleep to cool down
    print('Sleeping 5 sec')
    time.sleep(T)
    ps = dict()
    while n < N:
        if get_temp() > TL:
            x2 = []
            y2 = []
            for _ in range(int(sdt/dt)):
                x2.append(t)
                y2.append(get_temp())
                time.sleep(dt)
                t+=dt
            x2.append(t)
            y2.append(get_temp())
            xs2.append(x2)
            ys2.append(y2)
        else:
            x = []
            y = []
            while get_temp() < TL and n < N:
                for target, cmd in zip(targets, cmds):
                    if not target in ps:
                        ps[target] = None
                    if ps[target] is None or ps[target].poll() is not None:
                        if ps[target] is not None:
                            n+=1
                            print('Completed', n)
                            if n > N:
                                break
                        ps[target] = subprocess.Popen(cmd, env=env)
                x.append(t)
                y.append(get_temp())
                time.sleep(dt)
                t+=dt
            x.append(t)
            y.append(get_temp())
            xs.append(x)
            ys.append(y)
    plot_temp(xs, ys, xs2, ys2)

if __name__ == "__main__":
    run_sched()