JIT parallelize polynomial evaluation

mhostetter · Dec 8, 2022 · 4ca8b82 · 4ca8b82
1 parent 715da06
commit 4ca8b82
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/galois/_polys/_dense.py b/src/galois/_polys/_dense.py
@@ -428,11 +428,12 @@ def set_globals(self):
  MULTIPLY = self.field._multiply.ufunc_call_only
 
  _SIGNATURE = numba.types.FunctionType(int64[:](int64[:], int64[:]))
+ _PARALLEL = True
 
  @staticmethod
  def implementation(coeffs, values):
  y = np.zeros(values.size, dtype=values.dtype)
- for i in range(values.size):
+ for i in numba.prange(values.size):
  y[i] = coeffs[0]
  for j in range(1, coeffs.size):
  y[i] = ADD(coeffs[j], MULTIPLY(y[i], values[i]))