Skip to content

Commit c026ead

Browse files
committed
Add explicit C and C_VM modes
1 parent 2ea87b4 commit c026ead

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

pytensor/compile/mode.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -461,6 +461,9 @@ def clone(self, link_kwargs=None, optimizer="", **kwargs):
461461
RewriteDatabaseQuery(include=["fast_run", "py_only"]),
462462
)
463463

464+
C = Mode("c", "fast_run")
465+
C_VM = Mode("cvm", "fast_run")
466+
464467
NUMBA = Mode(
465468
NumbaLinker(),
466469
RewriteDatabaseQuery(
@@ -524,6 +527,8 @@ def clone(self, link_kwargs=None, optimizer="", **kwargs):
524527
predefined_modes = {
525528
"FAST_COMPILE": FAST_COMPILE,
526529
"FAST_RUN": FAST_RUN,
530+
"C": C,
531+
"C_VM": C_VM,
527532
"JAX": JAX,
528533
"NUMBA": NUMBA,
529534
"PYTORCH": PYTORCH,

pytensor/configdefaults.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,8 @@ def _filter_mode(val):
4545
"NanGuardMode",
4646
"FAST_COMPILE",
4747
"DEBUG_MODE",
48+
"CVM",
49+
"C",
4850
"JAX",
4951
"NUMBA",
5052
"PYTORCH",

0 commit comments

Comments
 (0)