llvm-project/offload/test/offloading/parallel_target_teams_reduction.cpp
Ethan Luis McDonough 8823448807
[Offload] Refactor offload test requirements (#95196)
Many tests in the `offload` project have requirements defined by which
targets are not supported rather than which platforms are supported.
This patch aims to streamline the requirement definitions by adding four
new feature tags: `host`, `gpu`, `amdgpu`, and `nvidiagpu`.
2024-06-29 00:56:18 -05:00

34 lines
684 B
C++

// RUN: %libomptarget-compilexx-run-and-check-generic
// RUN: %libomptarget-compileoptxx-run-and-check-generic
// FIXME: This is a bug in host offload, this should run fine.
// REQUIRES: gpu
#include <iostream>
#include <vector>
#define N 8
int main() {
std::vector<int> avec(N);
int *a = avec.data();
#pragma omp parallel for
for (int i = 0; i < N; i++) {
a[i] = 0;
#pragma omp target teams distribute parallel for reduction(+ : a[i])
for (int j = 0; j < N; j++)
a[i] += 1;
}
// CHECK: 8
// CHECK: 8
// CHECK: 8
// CHECK: 8
// CHECK: 8
// CHECK: 8
// CHECK: 8
// CHECK: 8
for (int i = 0; i < N; i++)
std::cout << a[i] << std::endl;
}