openmp/runtime/test/env/omp_num_threads_dev_all_fallback.c

// RUN: %libomp-compile && env OMP_NUM_THREADS_ALL=8 %libomp-run
//
// OpenMP 6.0: every non-host device falls through to `_ALL` when no
// `_DEV[_d]` is set.

#include <omp.h>
#include <stdio.h>
#include <string.h>

extern const char *__kmpc_get_device_env(const char *name, int device_id);

static int check(int device_id, const char *got, const char *expect) {
  if (!got || strcmp(got, expect) != 0) {
    fprintf(stderr, "FAIL: device_id=%d got '%s' expected '%s'\n", device_id,
            got ? got : "(null)", expect);
    return 1;
  }
  return 0;
}

int main(void) {
  int rc = 0;
  if (omp_get_max_threads() != 8) {
    fprintf(stderr, "FAIL: host omp_get_max_threads()=%d, expected 8\n",
            omp_get_max_threads());
    return 1;
  }
  rc |= check(0, __kmpc_get_device_env("OMP_NUM_THREADS", 0), "8");
  rc |= check(1, __kmpc_get_device_env("OMP_NUM_THREADS", 1), "8");
  rc |= check(2, __kmpc_get_device_env("OMP_NUM_THREADS", 2), "8");
  return rc;
}