Skip to content

Commit

Permalink
Merge pull request #73 from mapado/numba-use-numpy-kernel
Browse files Browse the repository at this point in the history
Use numpy kernel when using numba
  • Loading branch information
jdeniau committed Jan 16, 2024
2 parents 8f3dc9f + 09411a1 commit f3c3e90
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion haversine/haversine.py
Original file line number Diff line number Diff line change
Expand Up @@ -171,7 +171,7 @@ def _inverse_haversine_kernel(lat, lng, direction, d):
try:
import numba # type: ignore
if has_numpy:
_haversine_kernel_vector = numba.vectorize(fastmath=True)(_haversine_kernel)
_haversine_kernel_vector = numba.vectorize(fastmath=True)(_haversine_kernel_vector)
# Tuple output is not supported for numba.vectorize. Just jit the numpy version.
_inverse_haversine_kernel_vector = numba.njit(fastmath=True)(_inverse_haversine_kernel_vector)
_haversine_kernel = numba.njit(_haversine_kernel)
Expand Down

0 comments on commit f3c3e90

Please sign in to comment.