From 41540e4750893a287a14f1ee28f39877852865a2 Mon Sep 17 00:00:00 2001 From: Sergey Pokhodenko Date: Mon, 9 Nov 2020 22:42:22 +0300 Subject: [PATCH] Patch for lowering This change removes global initialization of lower_extensions with one functions and replaces with a list of functions. This change made in commit 947b407 by @reazulhoque. --- numba/core/lowering.py | 9 +-------- numba/parfors/parfor_lowering.py | 2 +- 2 files changed, 2 insertions(+), 9 deletions(-) diff --git a/numba/core/lowering.py b/numba/core/lowering.py index 1114ef65af0..c80b7911254 100644 --- a/numba/core/lowering.py +++ b/numba/core/lowering.py @@ -279,13 +279,6 @@ def debug_print(self, msg): class Lower(BaseLower): GeneratorLower = generators.GeneratorLower - def __init__(self, context, library, fndesc, func_ir, metadata=None): - BaseLower.__init__(self, context, library, fndesc, func_ir, metadata) - from numba.parfors.parfor_lowering import _lower_parfor_parallel - from numba.parfors import parfor - if parfor.Parfor not in lower_extensions: - lower_extensions[parfor.Parfor] = [_lower_parfor_parallel] - def pre_block(self, block): from numba.core.unsafe import eh @@ -452,7 +445,7 @@ def lower_inst(self, inst): else: for _class, func in lower_extensions.items(): if isinstance(inst, _class): - func[-1](self, inst) + func(self, inst) return raise NotImplementedError(type(inst)) diff --git a/numba/parfors/parfor_lowering.py b/numba/parfors/parfor_lowering.py index e559e8c017c..dc499498326 100644 --- a/numba/parfors/parfor_lowering.py +++ b/numba/parfors/parfor_lowering.py @@ -480,7 +480,7 @@ def _lower_parfor_parallel(lowerer, parfor): print("_lower_parfor_parallel done") # A work-around to prevent circular imports -#lowering.lower_extensions[parfor.Parfor] = _lower_parfor_parallel +lowering.lower_extensions[parfor.Parfor] = _lower_parfor_parallel def _create_shape_signature(