使用的是官方提供的代码SampleChapter3/3.1-vectoradd.cpp,具体如下
当前使用的环境是:
编译后用 sbatch vectoradd.slurm提交到计算节点上计算任务。其中vectoradd.slurm内容:
#SBATCH -p kshdtest
#SBATCH -N 1
#SBATCH -n 1
#SBATCH -o o.logs.vectoradd
#SBATCH -e e.logs.vectoradd
#SBATCH --ntasks-per-node=1
#SBATCH --cpus-per-task=1
#SBATCH --gres=dcu:1
#SBATCH --comment=addvector
module switch compiler/rocm/2.9 compiler/rocm/dtk-22.04.1
export HIP_KERNEL_PRINTF=1
rocminfo
# output information about this job and process CPU or GPU, node
date
echo ${SLURM_JOB_NODELIST}
sinfo -n ${SLURM_JOB_NODELIST}
jobid=squeue|tail -1|awk '{print $1;}'
scontrol show job $jobid
scontrol show node ${SLURM_JOB_NODELIST}
# run program
编译后用 sbatch vectoradd.slurm提交到计算节点上计算任务。其中vectoradd.slurm内容:
#SBATCH -p kshdtest
#SBATCH -N 1
#SBATCH -n 1
#SBATCH -o o.logs.vectoradd
#SBATCH -e e.logs.vectoradd
#SBATCH --ntasks-per-node=1
#SBATCH --cpus-per-task=1
#SBATCH --gres=dcu:1
#SBATCH --comment=addvector
module switch compiler/rocm/2.9 compiler/rocm/dtk-22.04.1
export HIP_KERNEL_PRINTF=1
rocminfo
# output information about this job and process CPU or GPU, node
date
echo ${SLURM_JOB_NODELIST}
sinfo -n ${SLURM_JOB_NODELIST}
jobid=squeue|tail -1|awk '{print $1;}'
scontrol show job $jobid
scontrol show node ${SLURM_JOB_NODELIST}
# run program
./vector_add_dcu > out