Default value to None

Author: reazulhoque

numba/parfors/parfor.py

@@ -411,7 +411,7 @@ def linspace_3(start, stop, num):
     else:
         raise ValueError("parallel linspace with types {}".format(args))
 
-replace_functions_map = {
+swap_functions_map = {
     ('argmin', 'numpy'): lambda r,a: argmin_parallel_impl,
     ('argmax', 'numpy'): lambda r,a: argmax_parallel_impl,
     ('min', 'numpy'): min_parallel_impl,
@@ -1387,14 +1387,16 @@ class PreParforPass(object):
     implementations of numpy functions if available.
     """
     def __init__(self, func_ir, typemap, calltypes, typingctx, options,
-                 swapped={}, replace_functions_map=replace_functions_map):
+                 swapped={}, replace_functions_map=None):
         self.func_ir = func_ir
         self.typemap = typemap
         self.calltypes = calltypes
         self.typingctx = typingctx
         self.options = options
         # diagnostics
         self.swapped = swapped
+        if replace_functions_map is None:
+            replace_functions_map = swap_functions_map
         self.replace_functions_map = replace_functions_map
         self.stats = {
             'replaced_func': 0,

numba/tests/test_parfors_passes.py

@@ -38,7 +38,7 @@ def __init__(self, typingctx, targetctx, args, test_ir):
 
 class BaseTest(TestCase):
     @classmethod
-    def _run_parfor(cls, test_func, args):
+    def _run_parfor(cls, test_func, args, swap_map=None):
         # TODO: refactor this with get_optimized_numba_ir() where this is
         #       copied from
         typingctx = typing.Context()
@@ -80,6 +80,7 @@ def _run_parfor(cls, test_func, args):
                 tp.state.typingctx,
                 options,
                 swapped=diagnostics.replaced_fns,
+                replace_functions_map=swap_map,
             )
             preparfor_pass.run()
 
@@ -109,9 +110,9 @@ def run_parfor_sub_pass(cls, test_func, args):
         return sub_pass
 
     @classmethod
-    def run_parfor_pre_pass(cls, test_func, args):
+    def run_parfor_pre_pass(cls, test_func, args, swap_map=None):
         tp, options, diagnostics, preparfor_pass = cls._run_parfor(
-            test_func, args
+            test_func, args, swap_map
         )
         return preparfor_pass
 
@@ -651,6 +652,20 @@ def test_impl(a):
         self.assertEqual(pre_pass.stats["replaced_dtype"], 0)
         self.run_parallel(test_impl, *args)
 
+    def test_replacement_map(self):
+        def test_impl(a):
+            return np.sum(a)
+
+        arr = np.arange(10)
+        args = (arr,)
+        argtypes = [typeof(x) for x in args]
+
+        swap_map = numba.parfors.parfor.swap_functions_map.copy()
+        swap_map.pop(("sum", "numpy"))
+        pre_pass = self.run_parfor_pre_pass(test_impl, argtypes, swap_map)
+        self.assertEqual(pre_pass.stats["replaced_func"], 0)
+        self.assertEqual(pre_pass.stats["replaced_dtype"], 0)
+        self.run_parallel(test_impl, *args)
 
 if __name__ == "__main__":
     unittest.main()