Console Output
+ source /etc/profile.d/larson-group.sh
++ export GIT_EDITOR=vi
++ GIT_EDITOR=vi
++ export SVN_EDITOR=vi
++ SVN_EDITOR=vi
++ export OMP_STACKSIZE=1048579
++ OMP_STACKSIZE=1048579
++ export LMOD_ROOT=/opt/lmod/
++ LMOD_ROOT=/opt/lmod/
++ source /opt/lmod//lmod/lmod/init/bash
+++ '[' -z '' ']'
+++ case "$-" in
+++ __lmod_vx=x
+++ '[' -n x ']'
+++ set +x
Shell debugging temporarily silenced: export LMOD_SH_DBG_ON=1 for this output (/usr/local/spack/opt/spack/linux-pop22-skylake_avx512/gcc-12.2.0/lmod-8.7.37-qorv6nkkjzyzntx6wbjupmoxrot26t6r/lmod/lmod/init/bash)
Shell debugging restarted
+++ unset __lmod_vx
+++ find /usr/local/spack/share/spack/lmod/linux-pop22-x86_64/Core -print -quit
++ export MODULEPATH=/usr/local/spack/share/spack/lmod/linux-pop22-x86_64/Core
++ MODULEPATH=/usr/local/spack/share/spack/lmod/linux-pop22-x86_64/Core
+ module load gcc netcdf-fortran
+ '[' -z '' ']'
+ case "$-" in
+ __lmod_sh_dbg=x
+ '[' -n x ']'
+ set +x
Shell debugging temporarily silenced: export LMOD_SH_DBG_ON=1 for Lmod's output
Lmod Warning:
-------------------------------------------------------------------------------
The following dependent module(s) are not currently loaded:
gcc-runtime/13.1.0-wiuv7wc (required by: mpich/4.2.1, hdf5/1.14.3-cb5lzz7,
netcdf-c/4.9.2, netcdf-fortran/4.5.3), glibc/2.35-tdr5czg (required by:
mpich/4.2.1, hdf5/1.14.3-cb5lzz7, netcdf-c/4.9.2, netcdf-fortran/4.5.3)
-------------------------------------------------------------------------------
Shell debugging restarted
+ unset __lmod_sh_dbg
+ return 0
+ ulimit -s 8388608
+ ./tests/run_jax_vs_fortran_cases.py -stats input/stats/standard_stats.in -j 4
==> CLUBB JAX profile: cpu (cpu)
==> Creating JAX virtualenv with existing Python 3.12
==> Installing uv 0.11.32 in /home/jenkins/workspace/clubb_jax_tests/.clubb-jax-tools/bin
downloading uv 0.11.32 x86_64-unknown-linux-gnu
installing to /home/jenkins/workspace/clubb_jax_tests/.clubb-jax-tools/bin
uv
uvx
everything's installed!
==> uv is ready: uv 0.11.32 (x86_64-unknown-linux-gnu)
Using CPython 3.12.3 interpreter at: /usr/bin/python3.12
Creating virtual environment at: .venv-jax
Activate with: source .venv-jax/bin/activate
==> Installing JAX 0.11.0 (cpu) and requirements from /home/jenkins/workspace/clubb_jax_tests/clubb_jax/requirements.txt
Using Python 3.12.3 environment at: .venv-jax
Resolved 15 packages in 401ms
Downloading scipy (33.7MiB)
Downloading numpy (15.9MiB)
Downloading pygments (1.2MiB)
Downloading jaxlib (83.2MiB)
Downloading netcdf4 (9.7MiB)
Downloading jax (3.1MiB)
Downloading cftime (1.6MiB)
Downloaded cftime
Downloaded pygments
Downloaded netcdf4
Downloaded jax
Downloaded numpy
Downloaded jaxlib
Downloaded scipy
Prepared 15 packages in 7.62s
Installed 15 packages in 67ms
+ certifi==2026.7.22
+ cftime==1.6.5
+ iniconfig==2.3.0
+ jax==0.11.0
+ jaxlib==0.11.0
+ ml-dtypes==0.6.0
+ netcdf4==1.7.4
+ numpy==2.5.3
+ opt-einsum==3.4.0
+ packaging==26.3
+ pluggy==1.6.0
+ pygments==2.21.0
+ pytest==9.1.1
+ scipy==1.18.1
+ tabulate==0.10.0
JAX environment ready: jax=0.11.0 jaxlib=0.11.0 backend=cpu devices=[CpuDevice(id=0)]
Python runtime: /home/jenkins/workspace/clubb_jax_tests/.venv-jax/bin/python
Running 13 case(s) x 1 flag set(s) = 13 run pair(s) with 4 worker(s)
Results root: /home/jenkins/workspace/clubb_jax_tests/output/tests/jax_driver_test_results
JAX output: /home/jenkins/workspace/clubb_jax_tests/output/tests/jax_driver_test_results/jax_output
Fortran output: /home/jenkins/workspace/clubb_jax_tests/output/tests/jax_driver_test_results/fortran_output
Flag sets: default
Stats: input/stats/standard_stats.in
Debug: (case default)
JAX accelerator: cpu
[match ] default bomex jax=444.5s f90=12.2s total=465.8s
[match ] default arm jax=436.3s f90=19.8s total=466.2s
[match ] default cobra jax=456.6s f90=15.7s total=483.0s
[match ] default atex jax=461.1s f90=22.4s total=494.8s
[match ] default fire jax=290.8s f90=6.0s total=303.8s
[match ] default dycoms2_rf01_fixed_sst jax=396.8s f90=8.7s total=418.0s
[match ] default dycoms2_rf01 jax=381.1s f90=39.9s total=437.0s
[match ] default dycoms2_rf02_nd jax=474.6s f90=8.9s total=497.4s
[match ] default gabls2 jax=175.8s f90=47.7s total=228.7s
[match ] default gabls3_night jax=156.2s f90=7.6s total=166.8s
[match ] default jun25_altocu jax=293.0s f90=12.9s total=313.7s
[match ] default neutral jax=381.4s f90=12.2s total=401.7s
[match ] default wangara jax=412.3s f90=11.2s total=431.2s
Summary:
{
"total_cases": 13,
"match": 13,
"diff": 0,
"jax_failed": 0,
"fortran_failed": 0,
"both_failed": 0,
"elapsed_s": 1514.7375192642212,
"final_bindiff_rc": 0,
"flag_sets": {
"default": {
"match": 13,
"diff": 0,
"jax_failed": 0,
"fortran_failed": 0,
"both_failed": 0
}
}
}
Detailed results: /home/jenkins/workspace/clubb_jax_tests/output/tests/jax_driver_test_results/case_compare_summary.json
Final bindiff log: /home/jenkins/workspace/clubb_jax_tests/output/tests/jax_driver_test_results/final_bindiff.log (rc=0)
