Skip to content

Add RIKYU support for SALMON - #75

Draft
william-dawson wants to merge 3 commits into
RIKEN-RCCS:developfrom
william-dawson:add-salmon-rikyu
Draft

Add RIKYU support for SALMON#75
william-dawson wants to merge 3 commits into
RIKEN-RCCS:developfrom
william-dawson:add-salmon-rikyu

Conversation

@william-dawson

Copy link
Copy Markdown

Summary

  • Adds RIKYU (GB200 NVL4, 4×B200/node) as a new target system for the SALMON benchmark, supporting 1–4 nodes (4 ranks/node).
  • build.sh: nvhpc/26.3 + OpenACC + cc100 (B200/Blackwell), with -DFORTRAN_COMPILER_HAS_MPI_VERSION3=OFF to work around the MPI_Iallreduce hang at >1 rank.
  • run.sh: per-rank GPU binding via wrapper.sh (CUDA_VISIBLE_DEVICES), UCX_IB_GPU_DIRECT_RDMA=no for 2+ nodes, and runtime patching of the &parallel namelist section to nproc_ob=N (orbital-only decomposition, no real-space grid split) — the verified GPU-optimal layout.
  • list.csv: 4 rows (1/2/3/4 nodes, 4 ranks/node, 1 thread, 30 min wall).
  • Input archive placed at /data1/rkp00012/CX_input/SALMON/SALMON.tar.gz (overridable via BK_SALMON_INPUT_ARCHIVE).

Depends on #74 for RIKYU system/queue config (gpu partition, account=rkp00012).

Test plan

  • bash programs/salmon/build.sh RIKYU succeeds on a RIKYU login/build node
  • bash programs/salmon/run.sh RIKYU 1 4 1 produces correct GS+RT results (1 node, 4 GPU)
  • bash programs/salmon/run.sh RIKYU 2 4 1 produces correct results (2 nodes, 8 GPU, exercises UCX fix)
  • Verify &parallel section is patched to nproc_ob=4 (1-node) / nproc_ob=8 (2-node) in the run logs

💘 Generated with Crush

Signed-off-by: William Dawson <william.dawson@riken.jp>
Signed-off-by: William Dawson <william.dawson@riken.jp>
@william-dawson
william-dawson marked this pull request as draft July 31, 2026 06:59
Build: nvhpc/26.3 + OpenACC + cc100 (B200/Blackwell), with
FORTRAN_COMPILER_HAS_MPI_VERSION3=OFF to work around the
MPI_Iallreduce hang at >1 rank.

Run: per-rank GPU binding via wrapper.sh (CUDA_VISIBLE_DEVICES),
UCX_IB_GPU_DIRECT_RDMA=no for 2+ nodes, and runtime patching of
the &parallel section to nproc_ob=N (orbital-only decomposition,
no real-space grid split) which is the verified GPU-optimal layout.

Depends on RIKEN-RCCS#74 for RIKYU system/queue config (gpu partition,
account=rkp00012).

Signed-off-by: William Dawson <william.dawson@riken.jp>

💘 Generated with Crush

Assisted-by: glm-5.2 via Crush <crush@charm.land>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant