burn_cpu.py

import numpy as np
import time
import math

def burn_cpu():
    a = np.random.rand(1024, 1024)
    b = np.random.rand(1024, 1024)
    tic = time.time()
    c = np.dot(a, b)
    toc = time.time()
    elapsed = (toc-tic)*1000
    seconds = 5
    times = int(math.floor((1000/elapsed) * seconds))
    print(f"elapsed: {elapsed:.1f} ms")
    print(f"running {times} times")

    tic = time.time()
    for _ in range(times):
        c = np.dot(a, b)
    toc = time.time()
    elapsed = (toc-tic)*1000
    print(f"elapsed: {elapsed:.1f} ms")

if __name__=="__main__":
    burn_cpu()