[UPDATE] Refactor after template update for RMA routines

This commit is contained in:
Stack-1
2026-05-26 13:15:31 +02:00
parent 4b8e4dd4a3
commit 29248d366f
971 changed files with 1844 additions and 1749 deletions
-50
View File
@@ -1,50 +0,0 @@
#!/usr/bin/env bash
#SBATCH --job-name=psb_spmv_overlap
#SBATCH --partition=boost_usr_prod
#SBATCH --time=01:00:00
#SBATCH --nodes=2
#SBATCH --ntasks=64
#SBATCH --ntasks-per-node=32
#SBATCH --cpus-per-task=1
#SBATCH --threads-per-core=1
#SBATCH --gpus-per-node=4
#SBATCH --export=ALL
#SBATCH -A CNHPC_1736213
#SBATCH --output=psb_spmv_overlap_%j.out
#SBATCH --error=psb_spmv_overlap_%j.err
set -euo pipefail
# Environment tuned like the existing comm test script.
export UCX_VFS_ENABLE=n
export UCX_VFS_USE_FUSE=n
export UCX_STATS_DEST=none
export UCX_LOG_LEVEL=error
export UCX_TLS=dc,sm,self
export UCX_NET_DEVICES=mlx5_0:1
export OMPI_MCA_coll=^hcoll,han
export OMPI_MCA_coll_hcoll_enable=0
export UCX_MEMTYPE_CACHE=n
export UCX_CLOSE_TIMEOUT=10s
EXEC=./test/comm/spmv/runs/spmv_overlap
IDIM_LIST=${IDIM_LIST:-"20 40 60 80 100 140 180 220 260 300"}
TIMES=${TIMES:-100}
BUILD_IF_MISSING=${BUILD_IF_MISSING:-1}
if [[ ! -x "$EXEC" ]]; then
echo "Executable not found: $EXEC" >&2
exit 1
fi
echo "Running SpMV overlap comm test"
echo " EXEC=$EXEC"
echo " IDIM_LIST=$IDIM_LIST"
echo " TIMES=$TIMES"
echo " BUILD_IF_MISSING=$BUILD_IF_MISSING"
for IDIM in $IDIM_LIST; do
echo ""
echo "=== Running IDIM=$IDIM TIMES=$TIMES ==="
IDIM="$IDIM" TIMES="$TIMES" srun --exclusive -N2 -n64 "$EXEC"
done
-6
View File
@@ -5,12 +5,6 @@ module psb_spmv_overlap_test
use psb_base_mod
use psb_util_mod
use psb_comm_schemes_mod, only: psb_comm_isend_irecv_, psb_comm_ineighbor_alltoallv_, &
& psb_comm_persistent_ineighbor_alltoallv_, psb_comm_rma_pull_, psb_comm_rma_push_, &
& psb_comm_handle_type
use psb_comm_baseline_mod, only: psb_comm_baseline_handle
use psb_comm_neighbor_impl_mod, only: psb_comm_neighbor_handle
use psb_comm_rma_mod, only: psb_comm_rma_handle
#ifdef PSB_HAVE_CUDA
use psb_cuda_mod
#endif