Files
llvm-project/offload/test/offloading/target-teams-atomic.c
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

51 lines
1.3 KiB
C

// Check that omp atomic is permitted and behaves when strictly nested within
// omp target teams. This is an extension to OpenMP 5.2 and is enabled by
// default.
// RUN: %libomptarget-compile-run-and-check-generic
#include <omp.h>
#include <stdbool.h>
#include <stdio.h>
#include <string.h>
// High parallelism increases our chances of detecting a lack of atomicity.
#define NUM_TEAMS_TRY 256
int main() {
// CHECK: update: num_teams=[[#NUM_TEAMS:]]{{$}}
// CHECK-NEXT: update: x=[[#NUM_TEAMS]]{{$}}
int x = 0;
int numTeams;
#pragma omp target teams num_teams(NUM_TEAMS_TRY) map(tofrom : x, numTeams)
{
#pragma omp atomic update
++x;
if (omp_get_team_num() == 0)
numTeams = omp_get_num_teams();
}
printf("update: num_teams=%d\n", numTeams);
printf("update: x=%d\n", x);
// CHECK-NEXT: capture: x=[[#NUM_TEAMS]]{{$}}
// CHECK-NEXT: capture: xCapturedCount=[[#NUM_TEAMS]]{{$}}
bool xCaptured[numTeams];
memset(xCaptured, 0, sizeof xCaptured);
x = 0;
#pragma omp target teams num_teams(NUM_TEAMS_TRY) map(tofrom : x, numTeams)
{
int v;
#pragma omp atomic capture
v = x++;
xCaptured[v] = true;
}
printf("capture: x=%d\n", x);
int xCapturedCount = 0;
for (int i = 0; i < numTeams; ++i) {
if (xCaptured[i])
++xCapturedCount;
}
printf("capture: xCapturedCount=%d\n", xCapturedCount);
return 0;
}