Numpy "benchmark"

numpy_benchmark.py

# benchmark.py
import time
import numpy as np

def benchmark():
    size = 2000
    a = np.random.random((size, size))
    b = np.random.random((size, size))

    start = time.time()
    c = np.dot(a, b)
    end = time.time()

    print(f"Matrix multiplication ({size}x{size}): {end-start:.2f}s")

    start = time.time()
    u, s, v = np.linalg.svd(a)
    end = time.time()

    print(f"SVD ({size}x{size}): {end-start:.2f}s")

if __name__ == "__main__":
    np.show_config()
    print("\n\n")
    benchmark()