-
Notifications
You must be signed in to change notification settings - Fork 115
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add workflow job on mac to check blas
- Loading branch information
1 parent
da9682f
commit 37c5713
Showing
1 changed file
with
82 additions
and
3 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -61,8 +61,86 @@ jobs: | |
python-version: ${{ matrix.python-version }} | ||
- uses: pre-commit/[email protected] | ||
|
||
test_mac: | ||
name: "Mac test py${{ matrix.python-version }} : fast-compile ${{ matrix.fast-compile }} : float32 ${{ matrix.float32 }} : ${{ matrix.part }}" | ||
needs: | ||
- changes | ||
- style | ||
runs-on: macos-latest | ||
if: ${{ needs.changes.outputs.changes == 'true' && needs.style.result == 'success' }} | ||
strategy: | ||
fail-fast: false | ||
matrix: | ||
os: ["macos-latest"] | ||
python-version: ["3.10", "3.12"] | ||
fast-compile: [0] | ||
float32: [0, 1] | ||
install-numba: [0] | ||
install-jax: [0] | ||
install-torch: [0] | ||
part: | ||
- "tests/tensor/test_blas.py tests/tensor/test_elemwise.py tests/tensor/test_math_scipy.py" | ||
exclude: | ||
- python-version: "3.10" | ||
float32: 1 | ||
steps: | ||
- uses: actions/checkout@v4 | ||
with: | ||
fetch-depth: 0 | ||
- name: Set up Python ${{ matrix.python-version }} | ||
uses: mamba-org/setup-micromamba@v1 | ||
with: | ||
environment-name: pytensor-test | ||
micromamba-version: "1.5.10-0" # until https://github.com/mamba-org/setup-micromamba/issues/225 is resolved | ||
init-shell: bash | ||
post-cleanup: "all" | ||
create-args: python=${{ matrix.python-version }} | ||
|
||
- name: Create matrix id | ||
id: matrix-id | ||
env: | ||
MATRIX_CONTEXT: ${{ toJson(matrix) }} | ||
run: | | ||
echo $MATRIX_CONTEXT | ||
export MATRIX_ID=`echo $MATRIX_CONTEXT | md5sum | cut -c 1-32` | ||
echo $MATRIX_ID | ||
echo "id=$MATRIX_ID" >> $GITHUB_OUTPUT | ||
- name: Install dependencies | ||
shell: micromamba-shell {0} | ||
run: | | ||
micromamba install --yes -q "python~=${PYTHON_VERSION}=*_cpython" numpy scipy pip graphviz cython pytest coverage pytest-cov pytest-benchmark pytest-mock libblas=*=*accelerate | ||
pip install pytest-sphinx | ||
pip install -e ./ | ||
micromamba list && pip freeze | ||
python -c 'import pytensor; print(pytensor.config.__str__(print_doc=False))' | ||
python -c 'import pytensor; assert pytensor.config.blas__ldflags != "", "Blas flags are empty"' | ||
env: | ||
PYTHON_VERSION: ${{ matrix.python-version }} | ||
|
||
- name: Run tests | ||
shell: micromamba-shell {0} | ||
run: | | ||
if [[ $FAST_COMPILE == "1" ]]; then export PYTENSOR_FLAGS=$PYTENSOR_FLAGS,mode=FAST_COMPILE; fi | ||
if [[ $FLOAT32 == "1" ]]; then export PYTENSOR_FLAGS=$PYTENSOR_FLAGS,floatX=float32; fi | ||
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 | ||
env: | ||
MATRIX_ID: ${{ steps.matrix-id.outputs.id }} | ||
PART: ${{ matrix.part }} | ||
FAST_COMPILE: ${{ matrix.fast-compile }} | ||
FLOAT32: ${{ matrix.float32 }} | ||
|
||
- name: Upload coverage file | ||
uses: actions/upload-artifact@v4 | ||
with: | ||
name: coverage-${{ steps.matrix-id.outputs.id }} | ||
path: coverage/coverage-${{ steps.matrix-id.outputs.id }}.xml | ||
|
||
test_ubuntu: | ||
name: "Test py${{ matrix.python-version }} : fast-compile ${{ matrix.fast-compile }} : float32 ${{ matrix.float32 }} : ${{ matrix.part }}" | ||
name: "Ubuntu test py${{ matrix.python-version }} : fast-compile ${{ matrix.fast-compile }} : float32 ${{ matrix.float32 }} : ${{ matrix.part }}" | ||
needs: | ||
- changes | ||
- style | ||
|
@@ -71,6 +149,7 @@ jobs: | |
strategy: | ||
fail-fast: false | ||
matrix: | ||
os: ["ubuntu-latest"] | ||
python-version: ["3.10", "3.12"] | ||
fast-compile: [0, 1] | ||
float32: [0, 1] | ||
|
@@ -249,10 +328,10 @@ jobs: | |
if: ${{ always() }} | ||
runs-on: ubuntu-latest | ||
name: "All tests" | ||
needs: [changes, style, test_ubuntu] | ||
needs: [changes, style, test_mac, test_ubuntu] | ||
steps: | ||
- name: Check build matrix status | ||
if: ${{ needs.changes.outputs.changes == 'true' && (needs.style.result != 'success' || needs.test_ubuntu.result != 'success') }} | ||
if: ${{ needs.changes.outputs.changes == 'true' && (needs.style.result != 'success' || needs.test_mac.result != 'success' || needs.test_ubuntu.result != 'success') }} | ||
run: exit 1 | ||
|
||
upload-coverage: | ||
|