doxygen/omp__num__threads__all__demo_8c_source.html

// RUN: %libomp-compile

// RUN: env OMP_NUM_THREADS_ALL=8 OMP_NUM_THREADS_DEV_0=128 %libomp-run \

// RUN:   | FileCheck %s

//

// Demo: host=8 (_ALL), device 0=128 (_DEV_0), other devices=8 (_ALL fallback).


#include <omp.h>

#include <stdio.h>


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


int main(void) {

  int host_max = omp_get_max_threads();

  printf("host omp_get_max_threads() = %d\n", host_max);

  for (int d = 0; d < 3; ++d) {

    const char *v = __kmpc_get_device_env("OMP_NUM_THREADS", d);

    printf("device %d resolved OMP_NUM_THREADS = %s\n", d, v ? v : "(default)");

  }

  return 0;

}


// CHECK: host omp_get_max_threads() = 8

// CHECK: device 0 resolved OMP_NUM_THREADS = 128

// CHECK: device 1 resolved OMP_NUM_THREADS = 8

// CHECK: device 2 resolved OMP_NUM_THREADS = 8

name
__itt_string_handle * name
Definition ittnotify.h:3305

d
void const char const char int ITT_FORMAT __itt_group_sync x void const char ITT_FORMAT __itt_group_sync s void ITT_FORMAT __itt_group_sync p void ITT_FORMAT p void ITT_FORMAT p no args __itt_suppress_mode_t unsigned int void size_t ITT_FORMAT d
Definition ittnotify_static.h:200

main
int main(void)
Definition omp_num_threads_all_demo.c:12

__kmpc_get_device_env
const char * __kmpc_get_device_env(const char *name, int device_id)
Definition kmp_device_env.cpp:311

omp_get_max_threads
int omp_get_max_threads()