在程序源文件中使用脚本变量OMP_NUM_THREADS

Using the script variable OMP_NUM_THREADS in the program source files

本文关键字：OMP 变量 NUM THREADS 脚本 程序 源文件      更新时间：2023-10-16

如果我在集群上运行c++代码，是否可以在我的程序中使用OMP_NUM_THREADS的值?例如，假设我有两个.cpp文件main.cpp和function .cpp，其中function .cpp是使用OpenMP并行编写的。我希望能够一次定义线程的数量(在下面的脚本中)，而不必在function .cpp中再次定义它。

#!/bin/bash
#PBS -S /bin/bash
#PBS -l walltime=00:10:00
#PBS -l select=1:ncpus=4:mem=2gb
#PBS -q QName
#PBS -N Name
#PBS -o Results/output.txt
#PBS -e Results/error.txt
#PBS -m abe -M email@address
module purge
module load intel-compiler/11.1.073
export OMP_NUM_THREADS=4
cd $WORKDIR
./myprog

您可以使用omp_set_num_threads()来设置程序中的线程数。
要使用OMP_NUM_THREADS外部指定的值，需要使用std::getenv从环境变量中读取它。请确保1)将字符串结果转换为数字，2)在未设置的情况下对值进行消毒。

伪代码看起来像这样: 
unsigned int thread_qty = std::max(atoi(std::getenv("OMP_NUM_THREADS")), 1);
omp_set_num_threads(thread_qty);

如果您将环境变量OMP_NUM_THREADS设置为某个值，并且从不触及代码中的线程数(例如通过omp_set_num_threads())，您的代码将使用

  • 禁用动态调整(OMP_DYNAMIC=FALSE/omp_set_dynamic(0)):代码将使用OMP_NUM_THREADS线程
  • 启用动态调整(OMP_DYNAMIC=TRUE/omp_set_dynamic(1)):代码将使用最多 OMP_NUM_THREADS线程(但可能使用更少)。

所以，只使用OMP_NUM_THREADS，不要在源代码中指定任何线程数

相关文章：