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-fq24mybyn2li6got2bxzk62ejh5d3p4z/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-nhc23hl,
netcdf-c/4.9.2-xduzp77, netcdf-fortran/4.5.3), glibc/2.35-tdr5czg (required by:
mpich/4.2.1, hdf5/1.14.3-nhc23hl, netcdf-c/4.9.2-xduzp77, netcdf-fortran/4.5.3)
-------------------------------------------------------------------------------
Shell debugging restarted
+ unset __lmod_sh_dbg
+ return 0
+ ulimit -s 8388608
+ python3 tests/run_jax_vs_fortran_cases.py -debug 0 -stats input/stats/standard_stats.in --max-iters 10
==> No Python 3.11+ found; uv will download 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)
Downloading cpython-3.12.13-linux-x86_64-gnu (download) (32.6MiB)
Downloaded cpython-3.12.13-linux-x86_64-gnu (download)
Using CPython 3.12.13
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.13 environment at: .venv-jax
Resolved 15 packages in 289ms
Downloading cftime (1.6MiB)
Downloading pygments (1.2MiB)
Downloading netcdf4 (9.7MiB)
Downloading jaxlib (83.2MiB)
Downloading scipy (33.7MiB)
Downloading jax (3.1MiB)
Downloading numpy (15.9MiB)
Downloaded cftime
Downloaded pygments
Downloaded netcdf4
Downloaded jax
Downloaded numpy
Downloaded scipy
Downloaded jaxlib
Prepared 15 packages in 2.06s
Installed 15 packages in 24ms
+ 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 (jax=0.11.0 jaxlib=0.11.0 backend=cpu devices=[CpuDevice(id=0)])
Running 13 case(s) with 8 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
Stats: input/stats/standard_stats.in
Debug: 0
JAX accelerator: cpu
[match ] arm jax=141.4s f90=1.0s total=145.6s
[match ] dycoms2_rf01_fixed_sst jax=141.5s f90=1.0s total=145.7s
[match ] bomex jax=141.8s f90=0.9s total=145.9s
[match ] fire jax=143.0s f90=1.0s total=146.9s
[match ] atex jax=143.0s f90=1.2s total=147.4s
[match ] dycoms2_rf01 jax=144.0s f90=1.9s total=149.9s
[match ] dycoms2_rf02_nd jax=146.2s f90=0.9s total=150.3s
[match ] cobra jax=147.7s f90=1.2s total=151.9s
[match ] neutral jax=94.4s f90=1.1s total=99.0s
[match ] gabls2 jax=95.9s f90=1.8s total=100.8s
[match ] jun25_altocu jax=99.7s f90=1.0s total=103.0s
[match ] wangara jax=99.5s f90=0.7s total=102.3s
[match ] gabls3_night jax=105.7s f90=0.6s total=108.0s
Summary:
{
"total_cases": 13,
"match": 13,
"diff": 0,
"jax_failed": 0,
"fortran_failed": 0,
"elapsed_s": 253.70539379119873
}
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)
