純粋なpythonのnumba.jitとnumba.cuda.jitから動的に/条件を選択して

Question

numba.cuda.jit、numba.jitおよびnone（pure python）から選択すると、デコレータのより良い動的制御を実現する最適な方法は何ですか？ [プロジェクトには10​​または100の機能があることに注意してください。これはすべての機能に適用しやすいはずです。 はnumbaウェブサイトの例です。

import numba as nb 
import numpy as np 

# global control of this --> @nb.jit or @nb.cuda.jit or none 
# some functions with @nb.jit or cuda.jit with kwargs like (nopython=True, **other_kwargs) 
def sum2d(arr): 
    M, N = arr.shape 
    result = 0.0 
    for i in range(M): 
     for j in range(N): 
      result += arr[i,j] 
    return result 


a = np.arange(81).reshape(9,9) 

sum2d(a) 

Answer 1

あなたは何かがより洗練された場合もありますが、比較的簡単な解決策は、設定に基づいてjitを再定義しています。例えば、

def _noop_jit(f=None, *args, **kwargs): 
    """ returns function unmodified, discarding decorator args""" 
    if f is None: 
     return lambda x: x 
    return f 

# some config flag 
if settings.PURE_PYTHON_MODE: 
    jit = _noop_jit 
else: # etc 
    from numba import jit 

@jit(nopython=True) 
def f(a): 
    return a + 1