Files
clang-p2996/offload/test/offloading/spmdization.c
Ethan Luis McDonough 8823448807 [Offload] Refactor offload test requirements (#95196)
Many tests in the `offload` project have requirements defined by which
targets are not supported rather than which platforms are supported.
This patch aims to streamline the requirement definitions by adding four
new feature tags: `host`, `gpu`, `amdgpu`, and `nvidiagpu`.
2024-06-29 00:56:18 -05:00

46 lines
1.3 KiB
C

// clang-format off
// RUN: %libomptarget-compileopt-generic
// RUN: env LIBOMPTARGET_INFO=16 \
// RUN: %libomptarget-run-generic 2>&1 | %fcheck-generic --check-prefixes=CHECK,SPMD
// RUN: %libomptarget-compileopt-generic -mllvm --openmp-opt-disable-spmdization
// RUN: env LIBOMPTARGET_INFO=16 \
// RUN: %libomptarget-run-generic 2>&1 | %fcheck-generic --check-prefixes=CHECK,GENERIC
// clang-format on
// REQUIRES: gpu
#include <omp.h>
#include <stdio.h>
__attribute__((weak)) void noop() {}
int main(void) {
int nthreads = 0, ip = 0, lvl = 0, alvl = 0, nested = 0, tid = 0, maxt = 0;
#pragma omp target map(from : nthreads, ip, lvl, alvl, nested, tid, maxt)
{
nthreads = omp_get_num_threads();
ip = omp_in_parallel();
lvl = omp_get_level();
alvl = omp_get_active_level();
nested = omp_get_nested();
tid = omp_get_thread_num();
maxt = omp_get_max_threads();
#pragma omp parallel
noop();
}
printf("NumThreads: %i, InParallel: %i, Level: %i, ActiveLevel: %i, Nested: %i, "
"ThreadNum: %i, MaxThreads: %i\n",
nthreads, ip, lvl, alvl, nested, tid, maxt);
// GENERIC: Generic mode
// SPMD: Generic-SPMD mode
// CHECK: NumThreads: 1
// CHECK: InParallel: 0
// CHECK: Level: 0
// CHECK: ActiveLevel: 0
// CHECK: Nested: 0
// CHECK: ThreadNum: 0
// CHECK: MaxThreads:
return 0;
}