mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-17 18:16:42 +00:00
8823448807
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`.
141 lines
4.0 KiB
C
141 lines
4.0 KiB
C
// RUN: %libomptarget-compile-run-and-check-generic
|
|
// REQUIRES: ompt
|
|
// REQUIRES: gpu
|
|
|
|
/*
|
|
* Example OpenMP program that registers EMI callbacks
|
|
*/
|
|
|
|
#include <assert.h>
|
|
#include <omp.h>
|
|
#include <stdio.h>
|
|
|
|
#include "callbacks.h"
|
|
#include "register_emi.h"
|
|
|
|
int main() {
|
|
int N = 100000;
|
|
|
|
int a[N];
|
|
int b[N];
|
|
|
|
int i;
|
|
|
|
for (i = 0; i < N; i++)
|
|
a[i] = 0;
|
|
|
|
for (i = 0; i < N; i++)
|
|
b[i] = i;
|
|
|
|
#pragma omp target parallel for
|
|
{
|
|
for (int j = 0; j < N; j++)
|
|
a[j] = b[j];
|
|
}
|
|
|
|
#pragma omp target teams distribute parallel for
|
|
{
|
|
for (int j = 0; j < N; j++)
|
|
a[j] = b[j];
|
|
}
|
|
|
|
int rc = 0;
|
|
for (i = 0; i < N; i++)
|
|
if (a[i] != b[i]) {
|
|
rc++;
|
|
printf("Wrong value: a[%d]=%d\n", i, a[i]);
|
|
}
|
|
|
|
if (!rc)
|
|
printf("Success\n");
|
|
|
|
return rc;
|
|
}
|
|
|
|
/// CHECK: Callback Init:
|
|
/// CHECK: Callback Load:
|
|
/// CHECK: Callback Target EMI: kind=1 endpoint=1
|
|
/// CHECK-NOT: code=(nil)
|
|
/// CHECK: code=[[CODE1:.*]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=1 optype=1
|
|
/// CHECK: code=[[CODE1]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=2 optype=1
|
|
/// CHECK-NOT: dest=(nil)
|
|
/// CHECK: code=[[CODE1]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=1 optype=2
|
|
/// CHECK: code=[[CODE1]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=2 optype=2
|
|
/// CHECK: code=[[CODE1]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=1 optype=1
|
|
/// CHECK: code=[[CODE1]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=2 optype=1
|
|
/// CHECK-NOT: dest=(nil)
|
|
/// CHECK: code=[[CODE1]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=1 optype=2
|
|
/// CHECK: code=[[CODE1]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=2 optype=2
|
|
/// CHECK: code=[[CODE1]]
|
|
/// CHECK: Callback Submit EMI: endpoint=1 req_num_teams=1
|
|
/// CHECK: Callback Submit EMI: endpoint=2 req_num_teams=1
|
|
/// CHECK: Callback DataOp EMI: endpoint=1 optype=3
|
|
/// CHECK: code=[[CODE1]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=2 optype=3
|
|
/// CHECK: code=[[CODE1]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=1 optype=3
|
|
/// CHECK: code=[[CODE1]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=2 optype=3
|
|
/// CHECK: code=[[CODE1]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=1 optype=4
|
|
/// CHECK: code=[[CODE1]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=2 optype=4
|
|
/// CHECK: code=[[CODE1]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=1 optype=4
|
|
/// CHECK: code=[[CODE1]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=2 optype=4
|
|
/// CHECK: code=[[CODE1]]
|
|
/// CHECK: Callback Target EMI: kind=1 endpoint=2
|
|
/// CHECK: code=[[CODE1]]
|
|
|
|
/// CHECK: Callback Target EMI: kind=1 endpoint=1
|
|
/// CHECK-NOT: code=(nil)
|
|
/// CHECK: code=[[CODE2:.*]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=1 optype=1
|
|
/// CHECK: code=[[CODE2]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=2 optype=1
|
|
/// CHECK-NOT: dest=(nil)
|
|
/// CHECK: code=[[CODE2]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=1 optype=2
|
|
/// CHECK: code=[[CODE2]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=2 optype=2
|
|
/// CHECK: code=[[CODE2]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=1 optype=1
|
|
/// CHECK: code=[[CODE2]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=2 optype=1
|
|
/// CHECK-NOT: dest=(nil)
|
|
/// CHECK: code=[[CODE2]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=1 optype=2
|
|
/// CHECK: code=[[CODE2]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=2 optype=2
|
|
/// CHECK: code=[[CODE2]]
|
|
/// CHECK: Callback Submit EMI: endpoint=1 req_num_teams=0
|
|
/// CHECK: Callback Submit EMI: endpoint=2 req_num_teams=0
|
|
/// CHECK: Callback DataOp EMI: endpoint=1 optype=3
|
|
/// CHECK: code=[[CODE2]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=2 optype=3
|
|
/// CHECK: code=[[CODE2]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=1 optype=3
|
|
/// CHECK: code=[[CODE2]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=2 optype=3
|
|
/// CHECK: code=[[CODE2]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=1 optype=4
|
|
/// CHECK: code=[[CODE2]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=2 optype=4
|
|
/// CHECK: code=[[CODE2]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=1 optype=4
|
|
/// CHECK: code=[[CODE2]]
|
|
/// CHECK: Callback DataOp EMI: endpoint=2 optype=4
|
|
/// CHECK: code=[[CODE2]]
|
|
/// CHECK: Callback Target EMI: kind=1 endpoint=2
|
|
/// CHECK: code=[[CODE2]]
|
|
/// CHECK: Callback Fini:
|