提交 789c0d17 authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Remove unused global_numba_func

上级 f6805683
...@@ -42,14 +42,6 @@ from pytensor.tensor.type import TensorType ...@@ -42,14 +42,6 @@ from pytensor.tensor.type import TensorType
from pytensor.tensor.type_other import MakeSlice, NoneConst from pytensor.tensor.type_other import MakeSlice, NoneConst
def global_numba_func(func):
"""Use to return global numba functions in numba_funcify_*.
This allows tests to remove the compilation using mock.
"""
return func
def numba_njit(*args, fastmath=None, **kwargs): def numba_njit(*args, fastmath=None, **kwargs):
kwargs.setdefault("cache", config.numba__cache) kwargs.setdefault("cache", config.numba__cache)
kwargs.setdefault("no_cpython_wrapper", True) kwargs.setdefault("no_cpython_wrapper", True)
......
...@@ -150,17 +150,8 @@ def eval_python_only(fn_inputs, fn_outputs, inputs, mode=numba_mode): ...@@ -150,17 +150,8 @@ def eval_python_only(fn_inputs, fn_outputs, inputs, mode=numba_mode):
else: else:
return lambda x: x return lambda x: x
def py_global_numba_func(func):
if hasattr(func, "py_func"):
return func.py_func
return func
mocks = [ mocks = [
mock.patch("numba.njit", njit_noop), mock.patch("numba.njit", njit_noop),
mock.patch(
"pytensor.link.numba.dispatch.basic.global_numba_func",
py_global_numba_func,
),
mock.patch( mock.patch(
"pytensor.link.numba.dispatch.basic.tuple_setitem", py_tuple_setitem "pytensor.link.numba.dispatch.basic.tuple_setitem", py_tuple_setitem
), ),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论