提交 30558ba5 authored 作者: ricardoV94's avatar ricardoV94 提交者: Ricardo Vieira

Make Numba the default linker

上级 0a10de29
...@@ -76,7 +76,7 @@ jobs: ...@@ -76,7 +76,7 @@ jobs:
strategy: strategy:
fail-fast: false fail-fast: false
matrix: matrix:
# default-mode: "CVM" is actually a no-op, to make sure we're testing default config settings # default-mode: "NUMBA" is actually a no-op, to make sure we're testing default config settings
default-mode: ["CVM", "NUMBA", "FAST_COMPILE"] default-mode: ["CVM", "NUMBA", "FAST_COMPILE"]
python-version: ["3.11", "3.14"] python-version: ["3.11", "3.14"]
os: ["ubuntu-latest"] os: ["ubuntu-latest"]
...@@ -192,7 +192,7 @@ jobs: ...@@ -192,7 +192,7 @@ jobs:
shell: micromamba-shell {0} shell: micromamba-shell {0}
run: | run: |
if [[ $DEFAULT_MODE == "FAST_COMPILE" ]]; then export PYTENSOR_FLAGS=$PYTENSOR_FLAGS,mode=FAST_COMPILE; fi if [[ $DEFAULT_MODE == "FAST_COMPILE" ]]; then export PYTENSOR_FLAGS=$PYTENSOR_FLAGS,mode=FAST_COMPILE; fi
if [[ $DEFAULT_MODE == "NUMBA" ]]; then export PYTENSOR_FLAGS=$PYTENSOR_FLAGS,linker=numba; fi if [[ $DEFAULT_MODE == "CVM" ]]; then export PYTENSOR_FLAGS=$PYTENSOR_FLAGS,linker=cvm; fi
export PYTENSOR_FLAGS=$PYTENSOR_FLAGS,warn__ignore_bug_before=all,on_opt_error=raise,on_shape_error=raise,gcc__cxxflags=-pipe export PYTENSOR_FLAGS=$PYTENSOR_FLAGS,warn__ignore_bug_before=all,on_opt_error=raise,on_shape_error=raise,gcc__cxxflags=-pipe
python -m pytest -r A --verbose --runslow --durations=50 --cov=pytensor/ --cov-report=xml:coverage/coverage-${MATRIX_ID}.xml --no-cov-on-fail $PART --benchmark-skip python -m pytest -r A --verbose --runslow --durations=50 --cov=pytensor/ --cov-report=xml:coverage/coverage-${MATRIX_ID}.xml --no-cov-on-fail $PART --benchmark-skip
env: env:
......
...@@ -512,7 +512,7 @@ def get_mode(orig_string): ...@@ -512,7 +512,7 @@ def get_mode(orig_string):
if upper_string == "FAST_RUN": if upper_string == "FAST_RUN":
linker = config.linker linker = config.linker
if linker == "auto": if linker == "auto":
return CVM if config.cxx else VM return NUMBA
return fast_run_linkers_to_mode[linker] return fast_run_linkers_to_mode[linker]
global _CACHED_RUNTIME_MODES global _CACHED_RUNTIME_MODES
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论