@@ -8,12 +8,18 @@ stages:
88
99variables :
1010 JUWELS_ACCOUNT : " cstma"
11+ JUWELS_PROJECT : " ccstma"
1112
1213
1314prepare_JUWELS :
1415 stage : benchmark
1516 rules :
1617 - if : $CI_COMMIT_MESSAGE !~ /.*\[CI-no-benchmarks\]/
18+ id_tokens :
19+ CI_JOB_JWT :
20+ aud : https://gitlab.jsc.fz-juelich.de
21+ SITE_ID_TOKEN :
22+ aud : https://gitlab.jsc.fz-juelich.de
1723 tags :
1824 - jacamar
1925 - juwels
@@ -29,8 +35,16 @@ prepare_JUWELS:
2935 - module load mpi4py
3036 - module load SciPy-Stack
3137 - module load CuPy
38+ - jutil env activate -p ${JUWELS_PROJECT}
39+ - python -m venv --clear $SCRATCH/.venv/pySDC
40+ - source $SCRATCH/.venv/pySDC/bin/activate
3241 - pip install -e .
3342 - pip install pytest-benchmark coverage
43+ - git submodule add -f https://github.com/brownbaerchen/mpi4py-fft.git
44+ - cd mpi4py-fft
45+ - git checkout cupy_implementation
46+ - FFTW_LIBRARY_DIR="/p/software/juwels/stages/2024/software/FFTW/3.3.10-GCC-12.3.0/lib64" pip install --force-reinstall .
47+ - cd ../
3448
3549
3650test_JUWELS :
@@ -39,6 +53,11 @@ test_JUWELS:
3953 - prepare_JUWELS
4054 rules :
4155 - if : $CI_COMMIT_MESSAGE !~ /.*\[CI-no-benchmarks\]/
56+ id_tokens :
57+ CI_JOB_JWT :
58+ aud : https://gitlab.jsc.fz-juelich.de
59+ SITE_ID_TOKEN :
60+ aud : https://gitlab.jsc.fz-juelich.de
4261 tags :
4362 - jacamar
4463 - juwels
@@ -63,11 +82,8 @@ test_JUWELS:
6382 - module load mpi4py
6483 - module load SciPy-Stack
6584 - module load CuPy
66- - git submodule add -f https://github.com/brownbaerchen/mpi4py-fft.git
67- - cd mpi4py-fft
68- - git checkout cupy_implementation
69- - FFTW_LIBRARY_DIR="/p/software/juwels/stages/2024/software/FFTW/3.3.10-GCC-12.3.0/lib64" pip install --force-reinstall -e .
70- - cd ../
85+ - jutil env activate -p ${JUWELS_PROJECT}
86+ - source $SCRATCH/.venv/pySDC/bin/activate
7187 script :
7288 # - touch benchmarks/output.json
7389 - echo $SYSTEMNAME
@@ -88,6 +104,11 @@ benchmark:
88104 when : manual
89105 tags :
90106 - docker
107+ id_tokens :
108+ CI_JOB_JWT :
109+ aud : https://gitlab.jsc.fz-juelich.de
110+ SITE_ID_TOKEN :
111+ aud : https://gitlab.jsc.fz-juelich.de
91112 rules :
92113 - if : $CI_COMMIT_MESSAGE !~ /.*\[CI-no-benchmarks\]/
93114 artifacts :
0 commit comments