Skip to content

Commit f680568

Browse files
committed
Remove numba__vectorize_target config
1 parent aa1b7c8 commit f680568

File tree

3 files changed

+0
-17
lines changed

3 files changed

+0
-17
lines changed

pytensor/configdefaults.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1099,12 +1099,6 @@ def add_scan_configvars():
10991099

11001100

11011101
def add_numba_configvars():
1102-
config.add(
1103-
"numba__vectorize_target",
1104-
("Default target for numba.vectorize."),
1105-
EnumStr("cpu", ["parallel", "cuda"], mutable=True),
1106-
in_c_key=False,
1107-
)
11081102
config.add(
11091103
"numba__fastmath",
11101104
("If True, use Numba's fastmath mode."),

pytensor/configparser.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,6 @@ class PyTensorConfigParser:
157157
scan__allow_gc: bool
158158
scan__allow_output_prealloc: bool
159159
# add_numba_configvars
160-
numba__vectorize_target: str
161160
numba__fastmath: bool
162161
numba__cache: bool
163162
# add_caching_dir_configvars

tests/link/numba/test_basic.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -769,16 +769,6 @@ def test_IfElse(inputs, cond_fn, true_vals, false_vals):
769769
compare_numba_and_py(inputs, out, test_values)
770770

771771

772-
@pytest.mark.xfail(reason="https://github.com/numba/numba/issues/7409")
773-
def test_config_options_parallel():
774-
x = pt.dvector()
775-
776-
with config.change_flags(numba__vectorize_target="parallel"):
777-
pytensor_numba_fn = function([x], pt.sum(x), mode=numba_mode)
778-
numba_mul_fn = pytensor_numba_fn.vm.jit_fn.py_func.__globals__["impl_sum"]
779-
assert numba_mul_fn.targetoptions["parallel"] is True
780-
781-
782772
def test_config_options_fastmath():
783773
x = pt.dvector()
784774

0 commit comments

Comments
 (0)