time.py

from numba import jit, cuda
import numpy as np
# to measure exec time
from tqdm import tqdm
from timeit import default_timer as timer


# normal function to run on cpu
def func(a):
    for i in range(100000000):
        a[i] += 1
        if i % 100 == 0:
            print(i)
    # function optimized to run on gpu


def func2(a):
    for i in tqdm(range(100000000)):
        a[i] += 1


if __name__ == "__main__":
    n = 100000000
    a = np.ones(n, dtype=np.float64)
    """
    start = timer()
    func(a)
    print("Prints", timer() - start)
"""
    start = timer()
    func2(a)
    print("tqdm", timer() - start)