Add malloc versions

Author: gjbex

source-code/cython/Primes/primes.py

@@ -6,7 +6,7 @@
 
 if __name__ == '__main__':
     arg_parser = ArgumentParser(description='compute primes')
-    arg_parser.add_argument('version', choices=['python', 'cython', 'pure_python'],
+    arg_parser.add_argument('version', choices=['python', 'cython', 'pure_python', 'cython_malloc', 'pure_malloc'],
                             help='version to run')
     arg_parser.add_argument('--n', type=int, default=10,
                             help='number of primes')
@@ -15,7 +15,11 @@
         from primes_vanilla import primes
     elif options.version == 'cython':
         from primes_cython import primes
+    elif options.version == 'cython_malloc':
+        from primes_malloc import primes
     elif options.version == 'pure_python':
         from primes_pure_python import primes
+    elif options.version == 'pure_malloc':
+        from primes_pure_malloc import primes
     results = primes(options.n)
     print(', '.join(map(str, results)))