提交 0f864a33 authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Consider config.numba__fastmath in cache key

上级 8b4c31ef
...@@ -478,7 +478,7 @@ def numba_funcify_ensure_cache(op, *args, **kwargs) -> tuple[Callable, str | Non ...@@ -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)", src=f"def {op_name}(*args): return jitable_func(*args)",
function_name=op_name, function_name=op_name,
global_env=globals() | {"jitable_func": jitable_func}, 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 return numba_njit(cached_func, cache=True), cache_key
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论