Skip to content

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-cascadelake/gcc-12.2.0/lmod-8.7.37-bi3kyxcdrfgw3y7vv2k7c5rjxg75qzju/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-gzedt3f (required by: mpich/4.2.1, hdf5/1.14.3-42x3zrv,
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
==> No compatible Python found for cpu; uv will download Python 3.12
==> Installing uv 0.11.32 in /home/jenkins/workspace/clubb_jax_tests_branch/.clubb-jax-tools/bin
downloading uv 0.11.32 x86_64-unknown-linux-gnu
installing to /home/jenkins/workspace/clubb_jax_tests_branch/.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_branch/clubb_jax/requirements.txt
Using Python 3.12.13 environment at: .venv-jax
Resolved 15 packages in 356ms
Downloading pygments (1.2MiB)
Downloading scipy (33.7MiB)
Downloading jax (3.1MiB)
Downloading numpy (15.9MiB)
Downloading netcdf4 (9.7MiB)
Downloading jaxlib (83.2MiB)
Downloading cftime (1.6MiB)
 Downloaded cftime
 Downloaded pygments
 Downloaded netcdf4
 Downloaded jax
 Downloaded numpy
 Downloaded scipy
 Downloaded jaxlib
Prepared 15 packages in 3.01s
Installed 15 packages in 55ms
 + 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
CLUBB JAX runtime:
  Environment: /home/jenkins/workspace/clubb_jax_tests_branch/.venv-jax (Python 3.12)
  Accelerator: cpu
  JAX: 0.11.0 (jaxlib 0.11.0)
  Device: cpu:0

Run settings:
  Output: /home/jenkins/workspace/clubb_jax_tests_branch/output/tests/jax_driver_test_results
  Statistics: input/stats/standard_stats.in
  Debug: case default
  Forwarded to both runs: -stats input/stats/standard_stats.in
  Columns: 4
  Workers: 4
  Comparison limits: absolute 1e-07; percentage 1e-07%

Cases and flag sets:
  20 cases x 1 flag set = 20 JAX/Fortran pairs
  Flag sets:
    default
  Cases:
    arm (360 iterations)
    atex (360 iterations)
    bomex (native duration)
    cobra (360 iterations)
    dycoms2_rf01 (native duration)
    dycoms2_rf01_fixed_sst (300 iterations)
    dycoms2_rf02_nd (native duration)
    fire (native duration)
    gabls2 (360 iterations)
    gabls3_night (360 iterations)
    jun25_altocu (180 iterations)
    neutral (native duration)
    wangara (native duration)
    dycoms2_rf02_do (native duration)
    dycoms2_rf02_ds (240 iterations)
    lba_kk (360 iterations)
      - input case: lba
      - microphysics_setting.microphys_scheme = "khairoutdinov_kogan"
      - microphysics_setting.lh_microphys_type = "disabled"
      - microphysics_setting.l_ice_microphys = .false.
      - microphysics_setting.l_graupel = .false.
    rico (380 iterations)
      - dt_main: 60 s
    lba (93 iterations)
      - percentage limit: 0.001%
      - microphysics_setting.lh_microphys_type = "disabled"
    clex9_oct14 (73 iterations)
      - percentage limit: 0.001%
    nov11_altocu (62 iterations)
      - percentage limit: 0.001%

[Status        ] Flag set                 Case                       Timesteps    JAX (s)  Fortran (s)  Total (s)
-----------------------------------------------------------------------------------------------------------------
[match         ] default                  bomex                      360 / 360      260.5          6.9      276.2
[match         ] default                  arm                        360 / 360      265.2          9.3      287.6
[match         ] default                  cobra                      360 / 360      270.4          8.6      289.5
[match         ] default                  atex                       360 / 360      274.9         12.3      304.9
[match         ] default                  fire                       180 / 180      148.4          3.2      155.6
[match         ] default                  dycoms2_rf01_fixed_sst     300 / 300      200.9          2.7      208.5
[match         ] default                  dycoms2_rf01               240 / 240      223.2         17.0      254.6
[match         ] default                  dycoms2_rf02_nd            360 / 360      246.1          4.4      255.3
[match         ] default                  gabls3_night               360 / 360       82.7          4.1       89.1
[match         ] default                  gabls2                     360 / 360      117.4         18.1      138.0
[match         ] default                  jun25_altocu               180 / 180      150.7          6.9      163.1
[match         ] default                  neutral                    360 / 360      235.0          7.0      248.0
[match         ] default                  wangara                    480 / 480      269.9          7.4      283.9
[match         ] default                  dycoms2_rf02_ds            240 / 240      382.5         20.0      408.5
[match         ] default                  rico                       380 / 380      293.9          8.2      304.7
[match         ] default                  dycoms2_rf02_do            360 / 360      555.0         31.0      595.3
[match         ] default                  lba_kk                     360 / 360      514.6         16.9      539.5
[match         ] default                  lba                          93 / 93      225.5          4.7      234.2
[match         ] default                  clex9_oct14                  73 / 73      167.3          4.2      175.3
[match         ] default                  nov11_altocu                 62 / 62      155.2          3.1      161.5

Flag set default:
  Match Diff Failed
  ----- ---- ------
     20    0      0

Total time: 23m 16.5s
Results JSON: /home/jenkins/workspace/clubb_jax_tests_branch/output/tests/jax_driver_test_results/case_compare_summary.json
Bindiff log: /home/jenkins/workspace/clubb_jax_tests_branch/output/tests/jax_driver_test_results/final_bindiff.log [passed]