Files
llvm-project/offload/test/mapping/prelock.cpp
Nick Sarnie 78ff5b55fd [offload][lit] Enable/disable tests on Level Zero when using DeviceRTL (#182128)
Since we can now build the DeviceRTL with SPIR-V, redo the
`XFAIL/UNSUPPORTED` specifications for the tests we see passing/failing
on the Level Zero backend with the DeviceRTL being used.

The tests marked `UNSUPPORTED` hang or sporadically fail and those are
tracked in https://github.com/llvm/llvm-project/issues/182119.

This change will allow us to enable CI testing with the DeviceRTL.

Here are the full test results with this change applied, running only
the `spirv64-intel` `check-offload` tests:

```
Total Discovered Tests: 453
  Unsupported      : 206 (45.47%)
  Passed           : 141 (31.13%)
  Expectedly Failed: 106 (23.40%)
```

31% is not a bad start.

---------

Signed-off-by: Nick Sarnie <nick.sarnie@intel.com>
2026-02-18 21:53:19 +00:00

66 lines
1.6 KiB
C++

// RUN: %libomptarget-compilexx-generic
// RUN: %libomptarget-run-generic %fcheck-generic
// REQUIRES: gpu
// UNSUPPORTED: nvidiagpu
// UNSUPPORTED: amdgpu
#include <cstdio>
#include <omp.h>
extern "C" {
void *llvm_omp_target_lock_mem(void *ptr, size_t size, int device_num);
void llvm_omp_target_unlock_mem(void *ptr, int device_num);
}
int main() {
int n = 100;
int *unlocked = new int[n];
for (int i = 0; i < n; i++)
unlocked[i] = i;
int *locked = (int *)llvm_omp_target_lock_mem(unlocked, n * sizeof(int),
omp_get_default_device());
if (!locked)
return 0;
#pragma omp target teams distribute parallel for map(tofrom : unlocked[ : n])
for (int i = 0; i < n; i++)
unlocked[i] += 1;
#pragma omp target teams distribute parallel for map(tofrom : unlocked[10 : 10])
for (int i = 10; i < 20; i++)
unlocked[i] += 1;
#pragma omp target teams distribute parallel for map(tofrom : locked[ : n])
for (int i = 0; i < n; i++)
locked[i] += 1;
#pragma omp target teams distribute parallel for map(tofrom : locked[10 : 10])
for (int i = 10; i < 20; i++)
locked[i] += 1;
llvm_omp_target_unlock_mem(unlocked, omp_get_default_device());
int err = 0;
for (int i = 0; i < n; i++) {
if (i < 10 || i > 19) {
if (unlocked[i] != i + 2) {
printf("Err at %d, got %d, expected %d\n", i, unlocked[i], i + 1);
err++;
}
} else if (unlocked[i] != i + 4) {
printf("Err at %d, got %d, expected %d\n", i, unlocked[i], i + 2);
err++;
}
}
// CHECK: PASS
if (err == 0)
printf("PASS\n");
return err;
}