tests/nproc/nproc-override.sh

   1 #!/bin/sh
   2 # Test the various OpenMP override options
   3
   4 # Copyright (C) 2017-2024 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
  20 print_ver_ nproc
  21
  22 unset OMP_NUM_THREADS
  23 unset OMP_THREADS_LIMIT
  24
  25 avail=$(nproc) || fail=1
  26 test $(($avail > 0)) || fail=1
  27
  28 #OMP_THREAD_LIMIT       OMP_NUM_THREADS     NPROC
  29 echo "\
  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
  47 while 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
  54 done
  55
  56 test -e failed && fail=1
  57
  58 Exit $fail