91 lines · c
1// clang-format off2// RUN: %libomptarget-compile-generic -DREQ=1 && %libomptarget-run-generic 2>&1 | %fcheck-generic -check-prefix=GOOD3// RUN: %libomptarget-compile-generic -DREQ=2 && not %libomptarget-run-generic 2>&1 | %fcheck-generic -check-prefix=BAD4// clang-format on5 6/*7 Test for the 'requires' clause check.8 When a target region is used, the requires flags are set in the9 runtime for the entire compilation unit. If the flags are set again,10 (for whatever reason) the set must be consistent with previously11 set values.12*/13#include <omp.h>14#include <stdio.h>15 16// ---------------------------------------------------------------------------17// Various definitions copied from OpenMP RTL18 19typedef struct {20 uint64_t Reserved;21 uint16_t Version;22 uint16_t Kind;23 uint32_t Flags;24 void *Address;25 char *SymbolName;26 uint64_t Size;27 uint64_t Data;28 void *AuxAddr;29} __tgt_offload_entry;30 31enum Flags {32 OMP_REGISTER_REQUIRES = 0x10,33};34 35typedef struct {36 void *ImageStart;37 void *ImageEnd;38 __tgt_offload_entry *EntriesBegin;39 __tgt_offload_entry *EntriesEnd;40} __tgt_device_image;41 42typedef struct {43 int32_t NumDeviceImages;44 __tgt_device_image *DeviceImages;45 __tgt_offload_entry *HostEntriesBegin;46 __tgt_offload_entry *HostEntriesEnd;47} __tgt_bin_desc;48 49void __tgt_register_lib(__tgt_bin_desc *Desc);50void __tgt_unregister_lib(__tgt_bin_desc *Desc);51 52// End of definitions copied from OpenMP RTL.53// ---------------------------------------------------------------------------54 55void run_reg_requires() {56 // Before the target region is registered, the requires registers the status57 // of the requires clauses. Since there are no requires clauses in this file58 // the flags state can only be OMP_REQ_NONE i.e. 1.59 60 // This is the 2nd time this function is called so it should print SUCCESS if61 // REQ is compatible with `1` and otherwise cause an error.62 __tgt_offload_entry entries[] = {63 {0, 0, 1, OMP_REGISTER_REQUIRES, NULL, "", 0, 1, NULL},64 {0, 0, 1, OMP_REGISTER_REQUIRES, NULL, "", 0, REQ, NULL}};65 __tgt_device_image image = {NULL, NULL, &entries[0], &entries[1] + 1};66 __tgt_bin_desc bin = {1, &image, &entries[0], &entries[1] + 1};67 68 __tgt_register_lib(&bin);69 70 printf("SUCCESS");71 72 __tgt_unregister_lib(&bin);73 74 // clang-format off75 // GOOD: SUCCESS76 // BAD: omptarget fatal error 2: '#pragma omp requires reverse_offload' not used consistently!77 // clang-format on78}79 80// ---------------------------------------------------------------------------81int main() {82 run_reg_requires();83 84// This also runs reg requires for the first time.85#pragma omp target86 {87 }88 89 return 0;90}91