aboutsummaryrefslogtreecommitdiff
path: root/openmp/runtime/test/env/omp_target_offload.c
blob: 4771e3c75dc7bf77a05a54036c231f48f847b710 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
// RUN: %libomp-compile-and-run
#include <string.h>
#include <stdlib.h>

enum kmp_target_offload_kind {
  tgt_disabled = 0,
  tgt_default = 1,
  tgt_mandatory = 2
};

extern int __kmpc_get_target_offload();
extern void kmp_set_defaults(char const *str);

const char *disabled_examples[] = {
    // Allowed inputs
    "disabled", "DISABLED", "Disabled", "dIsAbLeD", "DiSaBlEd"};

const char *default_examples[] = {
    // Allowed inputs
    "default", "DEFAULT", "Default", "deFAulT", "DEfaULt",
    // These should be changed to default (failed match)
    "mandatry", "defaults", "disable", "enabled", "mandatorynot"};

const char *mandatory_examples[] = {
    // Allowed inputs
    "mandatory", "MANDATORY", "Mandatory", "manDatoRy", "MANdATOry"};

// Return target-offload-var ICV
int get_target_offload_icv() {
#pragma omp parallel
  {}
  return __kmpc_get_target_offload();
}

int main() {
  int i;
  const char *omp_target_offload = "OMP_TARGET_OFFLOAD=";
  char buf[80];

  for (i = 0; i < sizeof(disabled_examples) / sizeof(char *); ++i) {
    strcpy(buf, omp_target_offload);
    strcat(buf, disabled_examples[i]);
    kmp_set_defaults(buf);
    if (tgt_disabled != get_target_offload_icv())
      return EXIT_FAILURE;
  }
  for (i = 0; i < sizeof(default_examples) / sizeof(char *); ++i) {
    strcpy(buf, omp_target_offload);
    strcat(buf, default_examples[i]);
    kmp_set_defaults(buf);
    if (tgt_default != get_target_offload_icv())
      return EXIT_FAILURE;
  }
  for (i = 0; i < sizeof(mandatory_examples) / sizeof(char *); ++i) {
    strcpy(buf, omp_target_offload);
    strcat(buf, mandatory_examples[i]);
    kmp_set_defaults(buf);
    if (tgt_mandatory != get_target_offload_icv())
      return EXIT_FAILURE;
  }

  return EXIT_SUCCESS;
}