We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 63d1506 commit ef149a1Copy full SHA for ef149a1
pytensor/link/numba/dispatch/basic.py
@@ -5,6 +5,7 @@
5
6
import numba
7
import numpy as np
8
+from numba import NumbaPerformanceWarning
9
from numba import njit as _njit
10
from numba.cpython.unsafe.tuple import tuple_setitem # noqa: F401
11
@@ -23,6 +24,16 @@
23
24
from pytensor.tensor.utils import hash_from_ndarray
25
26
27
+# Disable loud / incorrect warnings from Numba
28
+# https://github.com/numba/numba/issues/10086
29
+# TODO: Would be much better if we could disable only for our functions
30
+warnings.filterwarnings(
31
+ "ignore",
32
+ message="is faster on contiguous arrays",
33
+ category=NumbaPerformanceWarning,
34
+)
35
+
36
37
def numba_njit(
38
*args, fastmath=None, final_function: bool = False, **kwargs
39
) -> Callable:
0 commit comments