From 0fc7307c88247feb56001c9966ab7ccb21c20b97 Mon Sep 17 00:00:00 2001 From: Neil Gershenfeld <gersh@cba.mit.edu> Date: Thu, 6 Feb 2020 00:13:15 -0500 Subject: [PATCH] add Numba --- Python/numbapi.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100755 Python/numbapi.py diff --git a/Python/numbapi.py b/Python/numbapi.py new file mode 100755 index 0000000..4432dc3 --- /dev/null +++ b/Python/numbapi.py @@ -0,0 +1,27 @@ +# +# numbapi.py +# Neil Gershenfeld 2/6/20 +# calculation of pi by a Numba sum +# pi = 3.14159265358979323846 +# + +import time +from numba import jit + +NPTS = 100000000 +@jit(nopython=True) +def calc(): + a = 0.5 + b = 0.75 + c = 0.25 + pi = 0 + for i in range(1,(NPTS+1)): + pi += a/((i-b)*(i-c)) + return pi +start_time = time.time() +pi = calc() +end_time = time.time() +mflops = NPTS*5.0/(1.0e6*(end_time-start_time)) +print("NPTS = %d, pi = %f"%(NPTS,pi)) +print("time = %f, estimated MFlops = %f"%(end_time-start_time,mflops)) + -- GitLab