From e3439d092738a52452b95b31e9f4304abf289c5a Mon Sep 17 00:00:00 2001 From: Neil Gershenfeld <gersh@cba.mit.edu> Date: Thu, 2 Mar 2023 08:38:09 -0500 Subject: [PATCH] wip --- Python/cupi.py | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) create mode 100644 Python/cupi.py diff --git a/Python/cupi.py b/Python/cupi.py new file mode 100644 index 0000000..f079908 --- /dev/null +++ b/Python/cupi.py @@ -0,0 +1,28 @@ +# +# cupi.py +# Neil Gershenfeld 1/23/17 +# calculation of pi by a CuPy sum +# pi = 3.14159265358979323846 +# + +import numpy as np +import cupy as cp +import time + +NPTS = 100000000 +start_time = time.time() +i = np.arange(1,(NPTS+1),dtype=np.float64) +pi = np.sum(0.5/((i-0.75)*(i-.25))) +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 NumPy MFlops = %f"%(end_time-start_time,mflops)) + +NPTS = 100000000 +start_time = time.time() +i = cp.arange(1,(NPTS+1),dtype=cp.float64) +pi = cp.sum(0.5/((i-0.75)*(i-.25))) +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 CuPy MFlops = %f"%(end_time-start_time,mflops)) -- GitLab