Simple-Numba

Sat 17 May 2025

title: "Numba Simple" author: "Raja CSP Raman" date: 2019-05-07 description: "-" type: technical_note draft: false


from numba import njit, prange
import numpy as np

@njit(parallel=True)
def add(A):
    s = 0
    # Without "parallel=True" in the jit-decorator
    # the prange statement is equivalent to range
    for i in prange(A.shape[0]):
        s += A[i]
    return s
abc = add(np.array([1, 2, 3]))
abc
6

More on Numba

http://numba.pydata.org/numba-doc/latest/user/parallel.html#numba-parallel

http://numba.pydata.org/numba-doc/latest/index.html



Score: 5

Category: numba