提交 d4eff40d authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Add explicit C and C_VM modes

上级 7e5f7c85
...@@ -461,6 +461,9 @@ else: ...@@ -461,6 +461,9 @@ else:
RewriteDatabaseQuery(include=["fast_run", "py_only"]), RewriteDatabaseQuery(include=["fast_run", "py_only"]),
) )
C = Mode("c", "fast_run")
C_VM = Mode("cvm", "fast_run")
NUMBA = Mode( NUMBA = Mode(
NumbaLinker(), NumbaLinker(),
RewriteDatabaseQuery( RewriteDatabaseQuery(
...@@ -524,6 +527,8 @@ MLX = Mode( ...@@ -524,6 +527,8 @@ MLX = Mode(
predefined_modes = { predefined_modes = {
"FAST_COMPILE": FAST_COMPILE, "FAST_COMPILE": FAST_COMPILE,
"FAST_RUN": FAST_RUN, "FAST_RUN": FAST_RUN,
"C": C,
"C_VM": C_VM,
"JAX": JAX, "JAX": JAX,
"NUMBA": NUMBA, "NUMBA": NUMBA,
"PYTORCH": PYTORCH, "PYTORCH": PYTORCH,
......
...@@ -45,6 +45,8 @@ def _filter_mode(val): ...@@ -45,6 +45,8 @@ def _filter_mode(val):
"NanGuardMode", "NanGuardMode",
"FAST_COMPILE", "FAST_COMPILE",
"DEBUG_MODE", "DEBUG_MODE",
"CVM",
"C",
"JAX", "JAX",
"NUMBA", "NUMBA",
"PYTORCH", "PYTORCH",
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论