1#!/bin/sh
2# Test the various OpenMP override options
3
4# Copyright (C) 2017-2023 Free Software Foundation, Inc.
5
6# This program is free software: you can redistribute it and/or modify
7# it under the terms of the GNU General Public License as published by
8# the Free Software Foundation, either version 3 of the License, or
9# (at your option) any later version.
10
11# This program is distributed in the hope that it will be useful,
12# but WITHOUT ANY WARRANTY; without even the implied warranty of
13# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14# GNU General Public License for more details.
15
16# You should have received a copy of the GNU General Public License
17# along with this program.  If not, see <https://www.gnu.org/licenses/>.
18
19. "${srcdir=.}/tests/init.sh"; path_prepend_ ./src
20print_ver_ nproc
21
22unset OMP_NUM_THREADS
23unset OMP_THREADS_LIMIT
24
25avail=$(nproc) || fail=1
26test $(($avail > 0)) || fail=1
27
28#OMP_THREAD_LIMIT       OMP_NUM_THREADS     NPROC
29echo "\
30 -                      -                   $avail
31 1                      -                   1
32 1                      0                   1
33 -                      0                   $avail
34 -                      2,2,1               2
35 -                      2,ignored           2
36 -                      2bad                $avail
37 -                      -2                  $avail
38 1                      2,2,1               1
39 0                      2,2,1               2
40 1bad                   2,2,1               2
41 1bad                   $(($avail+1)),2,1   $(($avail+1))
42 1                      $(($avail+1))       1
43 $(($avail+2))          $(($avail+1))       $(($avail+1))
44 $(($avail+1))          $(($avail+2))       $(($avail+1))
45 -                      $(($avail+1))       $(($avail+1))" |
46
47while read OMP_THREAD_LIMIT OMP_NUM_THREADS NPROC; do
48  test $OMP_THREAD_LIMIT = '-' && unset OMP_THREAD_LIMIT
49  test $OMP_NUM_THREADS = '-' && unset OMP_NUM_THREADS
50  export OMP_THREAD_LIMIT
51  export OMP_NUM_THREADS
52  test $(nproc) = $NPROC ||
53    echo "[$OMP_THREAD_LIMIT] [$OMP_NUM_THREADS]" >> failed
54done
55
56test -e failed && fail=1
57
58Exit $fail
59