In [1]:
import numpy as np
import time as time
import matplotlib.pyplot as plt
In [2]:
#nlist = 10**np.array([1,2,3,4,5,6,7,8])
nlist = np.arange(64,1e5,64,dtype=int)
In [3]:
tavg = []
tmax = []
tmin = []

for n in nlist:
    v = np.ones((n,),dtype=np.float16)
    w = np.ones((n,),dtype=np.float16)
    u = v + w
    
    m=5
    ts = np.zeros((m,))
    for i in range(m):
        t = time.time()
        u = v + w
        ts[i] = time.time() - t
    tavg.append(ts.mean())
    tmax.append(ts.max())
    tmin.append(ts.min())
In [5]:
plt.loglog(nlist, tavg)
plt.loglog(nlist, tmax)
plt.loglog(nlist, tmin)
plt.grid(True)
In [ ]: