OpenMP Programming : How to specify the number of threads as a command line option

Question

I am running a local blastx server. One of the command line options is -num_threads. Looking at the executable, blastx, thinking it may be a shell script that sets OMP_NUM_THREADS, it turns out that it is in machine code. I am assuming (possibly incorrectly) that it is an OpenMP application and this got me thinking.

Question : is it possible to change the number of OpenMP threads as a command line option, as opposed to using the environmental variable OMP_NUM_THREADS?

What you want is `omp_set_num_threads` along with some other things. This may help: http://stackoverflow.com/questions/11095309/openmp-set-num-threads-is-not-working — Craig Estey, Dec 21 '15 at 21:59

score 12 · Accepted Answer · answered Dec 22 '15 at 07:15

Using OpenMP, you have basically 3 different ways of specifying the number of threads to use in a parallel region:

The most commonly used one is the environment variable OMP_NUM_THREADS which needs to be set in the code's environment prior to running it for being effective;
The function omp_set_num_threads(), to be called before reaching a parallel region; and
The optional num_threads() clause of the parallel directive.

The relative priorities of these are defined in great details by the standard but pretty much boil down to num_threads() taking precedence over omp_set_num_threads(), which itself takes precedence over OMP_NUM_THREADS.

So now, if you want to have your code defining the number of OpenMP threads to use as a command line option, what you need is:

To parse you command line, either by hand, or using a function like getopt, and to store the value you read in a variable; and
To use this value in either a call to omp_set_num_threads() or as a parameter to the num_threads() clause. Either of the two will take precedence to the possible value set for OMP_NUM_THREADS.

Jeff Hammond · Answer 2 · 2015-12-21T22:19:15.817

1

OMP_NUM_THREADS is parsed by the program at runtime, so it already does what you want.

Setting it at compile time has no effect (unless you specifically design your build system to use it).

Because you export this environment variable, it's there at runtime as well. That's why you think it is doing something when you compile.

edited Dec 21 '15 at 22:19

answered Dec 21 '15 at 22:15

Jeff Hammond

5,374
3
28
45

Yes, of course. I fixed the wording of the question appropriately – irritable_phd_syndrome Dec 21 '15 at 22:17
You can use something like getopt or equivalent and then set as noted by other comments. But what do this gain you? – Jeff Hammond Dec 21 '15 at 22:20

OpenMP Programming : How to specify the number of threads as a command line option

2 Answers2