-
Notifications
You must be signed in to change notification settings - Fork 2
68 lines (58 loc) · 1.6 KB
/
tests.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
name: tests
on:
push:
branches:
- main
pull_request: null
jobs:
tests:
name: tests
runs-on: "ubuntu-latest"
steps:
- uses: n1hility/cancel-previous-runs@v2
with:
token: ${{ secrets.GITHUB_TOKEN }}
- uses: actions/checkout@v3
- uses: conda-incubator/setup-miniconda@v2
with:
python-version: 3.9
channels: conda-forge,defaults
channel-priority: strict
show-channel-urls: true
environment-file: environment.yaml
miniforge-version: latest
miniforge-variant: Mambaforge
use-mamba: true
- name: configure conda and install code
shell: bash -l {0}
run: |
python -m pip install -e .
mamba install -y -q \
flake8 \
pytest \
pytest-mpi \
setuptools \
pip \
poetry
- name: lint
shell: bash -l {0}
run: |
flake8 diffsmhm
- name: test
shell: bash -l {0}
run: |
export JAX_ENABLE_X64=True
export NUMBA_ENABLE_CUDASIM=1
pytest -vvs diffsmhm
- name: test mpi 8
shell: bash -l {0}
run: |
export JAX_ENABLE_X64=True
export NUMBA_ENABLE_CUDASIM=1
mpirun -n 8 --oversubscribe python -m pytest -vv --only-mpi diffsmhm
- name: test mpi 12
shell: bash -l {0}
run: |
export JAX_ENABLE_X64=True
export NUMBA_ENABLE_CUDASIM=1
mpirun -n 12 --oversubscribe python -m pytest -vv --only-mpi diffsmhm