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 64eeee1 commit a772728Copy full SHA for a772728
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
@@ -24,6 +25,16 @@
24
25
from pytensor.typed_list import TypedListType
26
27
28
+# Disable loud / incorrect warnings from Numba
29
+# https://github.com/numba/numba/issues/10086
30
+# TODO: Would be much better if we could disable only for our functions
31
+warnings.filterwarnings(
32
+ "ignore",
33
+ message="is faster on contiguous arrays",
34
+ category=NumbaPerformanceWarning,
35
+)
36
+
37
38
def numba_njit(
39
*args, fastmath=None, final_function: bool = False, **kwargs
40
) -> Callable:
0 commit comments