omp_num_threads_host_vs_device.c

Go to the documentation of this file.

// RUN: %libomp-compile && \
// RUN:   env OMP_NUM_THREADS=4 OMP_NUM_THREADS_ALL=8 \
// RUN:   OMP_NUM_THREADS_DEV_0=128 %libomp-run
//
// Host ICV via omp_get_max_threads; devices via __kmpc_get_device_env.
 
#include <omp.h>
#include <stdio.h>
#include <string.h>
 
extern const char *__kmpc_get_device_env(const char *name, int device_id);
 
int main(void) {
  if (omp_get_max_threads() != 4)
    return 1;
 
  // Device 0 uses _DEV_0; device 1 falls through to _ALL.
  const char *dev0 = __kmpc_get_device_env("OMP_NUM_THREADS", 0);
  if (dev0 == NULL || strcmp(dev0, "128") != 0)
    return 2;
  const char *dev1 = __kmpc_get_device_env("OMP_NUM_THREADS", 1);
  if (dev1 == NULL || strcmp(dev1, "8") != 0)
    return 3;
 
  return 0;
}