TheRock/compiler/amd-llvm/offload/test/offloading/thread_limit.c

// clang-format off
// RUN: %libomptarget-compile-generic
// RUN: env LIBOMPTARGET_INFO=16 \
// RUN:   %libomptarget-run-generic 2>&1 | %fcheck-generic --check-prefix=DEFAULT

// UNSUPPORTED: nvptx64-nvidia-cuda
// UNSUPPORTED: nvptx64-nvidia-cuda-LTO
// REQUIRES: gpu
// XFAIL: intelgpu

// clang-format on
int main() {
  int n = 1 << 20;
  int th = 12;
  int te = n / th;

// DEFAULT: 12 (MaxFlatWorkGroupSize:
#pragma omp target
#pragma omp teams loop num_teams(te), thread_limit(th)
  for (int i = 0; i < n; i++) {
  }

// DEFAULT: 13 (MaxFlatWorkGroupSize:
  #pragma omp target
  #pragma omp teams distribute parallel for simd num_teams(te), thread_limit(th+1) simdlen(64)
  for(int i = 0; i < n; i++) {
  }

// DEFAULT: 128 (MaxFlatWorkGroupSize:
#pragma omp target teams distribute parallel for thread_limit(128)
  for (int i = 0; i < n; i++) {
  }

// DEFAULT: 512 (MaxFlatWorkGroupSize:
#pragma omp target teams distribute parallel for thread_limit(512)
  for (int i = 0; i < n; i++) {
  }

// DEFAULT: 1024 (MaxFlatWorkGroupSize:
#pragma omp target teams distribute parallel for thread_limit(1024)
  for (int i = 0; i < n; i++) {
  }

// DEFAULT: 128 (MaxFlatWorkGroupSize:
#pragma omp target teams distribute parallel for num_threads(128)
  for (int i = 0; i < n; i++) {
  }

// DEFAULT: 512 (MaxFlatWorkGroupSize:
#pragma omp target teams distribute parallel for num_threads(512)
  for (int i = 0; i < n; i++) {
  }

// DEFAULT: 1024 (MaxFlatWorkGroupSize:
#pragma omp target teams distribute parallel for num_threads(1024)
  for (int i = 0; i < n; i++) {
  }

// DEFAULT: 64 (MaxFlatWorkGroupSize:
#pragma omp target teams distribute parallel for thread_limit(64)              \
    num_threads(128)
  for (int i = 0; i < n; i++) {
  }

// DEFAULT: 64 (MaxFlatWorkGroupSize:
#pragma omp target teams distribute parallel for thread_limit(128)             \
    num_threads(64)
  for (int i = 0; i < n; i++) {
  }

// DEFAULT: 512 (MaxFlatWorkGroupSize:
#pragma omp target teams distribute parallel for thread_limit(512)             \
    num_threads(1024)
  for (int i = 0; i < n; i++) {
  }

// DEFAULT: 512 (MaxFlatWorkGroupSize:
#pragma omp target teams distribute parallel for thread_limit(1024)            \
    num_threads(512)
  for (int i = 0; i < n; i++) {
  }

  return 0;
}