Skip to content

Commit

Permalink
Re-add numba wrapper that was mistakenly removed
Browse files Browse the repository at this point in the history
  • Loading branch information
moble committed Apr 18, 2018
1 parent 13aef40 commit c85a3d6
Showing 1 changed file with 49 additions and 0 deletions.
49 changes: 49 additions & 0 deletions numba_wrapper.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,49 @@
# Copyright (c) 2017, Michael Boyle
# See LICENSE file for details: <https://github.com/moble/quaternion/blob/master/LICENSE>

from __future__ import division, print_function, absolute_import

## Allow the code to function without numba, but discourage it
## strongly.
try:
from numbapro import njit, jit, vectorize, int64, float64, complex128
from numba.utils import IS_PY3

GOT_NUMBA = True
except ImportError:
try:
from numba import njit, jit, vectorize, int64, float64, complex128
from numba.utils import IS_PY3

GOT_NUMBA = True
except ImportError:
import warnings
import sys

warning_text = \
"\n\n" + "!" * 53 + "\n" + \
"Could not import from either numbapro or numba.\n" + \
"This means that the code will run MUCH more slowly.\n" + \
"You probably REALLY want to install numba / numbapro." + \
"\n" + "!" * 53 + "\n"
warnings.warn(warning_text)

def _identity_decorator_outer(*args, **kwargs):
def _identity_decorator_inner(fn):
return fn

return _identity_decorator_inner

njit = _identity_decorator_outer
jit = _identity_decorator_outer
vectorize = _identity_decorator_outer
int64 = int
float64 = float
complex128 = complex
IS_PY3 = (sys.version_info[:2] >= (3, 0))
GOT_NUMBA = False

if IS_PY3:
xrange = range
else:
xrange = xrange

0 comments on commit c85a3d6

Please sign in to comment.