Final code for Cython threading.

Author: mikeckennedy

src/11-cython/perf/compute_cython.py

@@ -9,15 +9,15 @@ def main():
 
     t0 = datetime.datetime.now()
 
-    # do_math(num=30000000)
+    # math_core.do_math(num=30_000_000)
     print("Doing math on {:,} processors.".format(multiprocessing.cpu_count()))
 
     processor_count = multiprocessing.cpu_count()
     threads = []
     for n in range(1, processor_count + 1):
         threads.append(Thread(target=math_core.do_math,
-                              args=(30_000_000 * (n - 1) / processor_count,
-                                    30_000_000 * n / processor_count),
+                              args=(300_000 * (n - 1) / processor_count,
+                                    300_000 * n / processor_count),
                               daemon=True)
                        )
 
@@ -27,7 +27,7 @@ def main():
     dt = datetime.datetime.now() - t0
     print("Done in {:,.2f} sec. (factor: {:,.2f}x)".format(
         dt.total_seconds(),
-        8.54/dt.total_seconds())
+        .09/dt.total_seconds())
     )
 
 

src/11-cython/perf/compute_it.py

@@ -7,7 +7,7 @@ def main():
 
     t0 = datetime.datetime.now()
 
-    do_math(num=30_000_000)
+    do_math(num=300_000)
 
     dt = datetime.datetime.now() - t0
     print("Done in {:,.2f} sec.".format(dt.total_seconds()))

src/11-cython/perf/math_core.pyx

@@ -1,8 +1,12 @@
-import math
+from libc.math cimport sqrt
 
-def do_math(start=0, num=10):
-    pos = start
-    k_sq = 1000 * 1000
-    while pos < num:
-        pos += 1
-        math.sqrt((pos - k_sq) * (pos - k_sq))
+import cython
+
+def do_math(start: cython.int = 0, num: cython.int = 10):
+    pos: cython.int = start
+    k_sq: cython.int = 1000 * 1000
+
+    with nogil:
+        while pos < num:
+            pos += 1
+            sqrt((pos - k_sq) * (pos - k_sq))