Skip to content

Commit 2677a6e

Browse files
committed
Add more linker options to config
1 parent 92c3b49 commit 2677a6e

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
@@ -369,11 +369,21 @@ def add_compile_configvars():
369369

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

0 commit comments

Comments
 (0)