Merge pull request #163 from bashtage/fastmath

PERF: Enable fastmath for numba

Author: bashtage

arch/compat/numba.py

@@ -1,9 +1,18 @@
 from __future__ import absolute_import, division
+import functools
 
 try:
     from numba import jit
+    try:
+        def f(x, y):
+            return x + y
+        fjit = jit(f, nopython=True, fastmath=True)
+        fjit(1.0, 2.0)
+        jit = functools.partial(jit, nopython=True, fastmath=True)
+    except KeyError:
+        jit = functools.partial(jit, nopython=True)
 except ImportError:
-    def jit(func):
+    def jit(func, *args, **kwargs):
         def wrapper(*args, **kwargs):
             return func(*args, **kwargs)