Consider config.numba__fastmath in cache key

ricardoV94 · ricardoV94 · commit 0f864a33958e · 2026-01-02T17:20:52.000+01:00
diff --git a/pytensor/link/numba/dispatch/basic.py b/pytensor/link/numba/dispatch/basic.py
@@ -478,7 +478,7 @@ def numba_funcify_ensure_cache(op, *args, **kwargs) -> tuple[Callable, str | Non
             src=f"def {op_name}(*args): return jitable_func(*args)",
             function_name=op_name,
             global_env=globals() | {"jitable_func": jitable_func},
-            cache_key=cache_key,
+            cache_key=f"{cache_key}_fastmath{int(config.numba__fastmath)}",
         )
         return numba_njit(cached_func, cache=True), cache_key
 

Original file line number	Diff line number	Diff line change
`@@ -478,7 +478,7 @@ def numba_funcify_ensure_cache(op, args, *kwargs) -> tuple[Callable, str \| Non`
`478`	`478`	`src=f"def {op_name}(args): return jitable_func(args)",`
`479`	`479`	`function_name=op_name,`
`480`	`480`	`global_env=globals() \| {"jitable_func": jitable_func},`
`481`		`- cache_key=cache_key,`
	`481`	`+ cache_key=f"{cache_key}_fastmath{int(config.numba__fastmath)}",`
`482`	`482`	`)`
`483`	`483`	`return numba_njit(cached_func, cache=True), cache_key`
`484`	`484`