Skip to content

Commit 3a352ca

Browse files
committed
Add more linker options to config
1 parent 7779b07 commit 3a352ca

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

pytensor/configdefaults.py

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -370,11 +370,21 @@ def add_compile_configvars():
370370

371371
if rc == 0 and config.cxx != "":
372372
# Keep the default linker the same as the one for the mode FAST_RUN
373-
linker_options = ["c|py", "py", "c", "c|py_nogc", "vm", "vm_nogc", "cvm_nogc"]
373+
linker_options = [
374+
"c|py",
375+
"py",
376+
"c",
377+
"c|py_nogc",
378+
"vm",
379+
"vm_nogc",
380+
"cvm_nogc",
381+
"jax",
382+
"numba",
383+
]
374384
else:
375385
# g++ is not present or the user disabled it,
376386
# linker should default to python only.
377-
linker_options = ["py", "vm_nogc"]
387+
linker_options = ["py", "vm", "vm_nogc", "numba", "jax"]
378388
if type(config).cxx.is_default:
379389
# If the user provided an empty value for cxx, do not warn.
380390
_logger.warning(

0 commit comments

Comments
 (0)