xref: /llvm-project/llvm/test/Transforms/OpenMP/values_in_offload_arrays.ll (revision aa8e9fac2a14bc7f0eb9aac20994d0c310d1ae92)
1; RUN: opt -S -passes=openmp-opt-cgscc -aa-pipeline=basic-aa -openmp-hide-memory-transfer-latency -debug-only=openmp-opt < %s 2>&1 | FileCheck %s
2; REQUIRES: asserts
3
4target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"
5
6@.__omp_offloading_heavyComputation.region_id = weak constant i8 0
7@.offload_maptypes. = private unnamed_addr constant [2 x i64] [i64 35, i64 35]
8
9%struct.ident_t = type { i32, i32, i32, i32, ptr }
10
11@.str = private unnamed_addr constant [23 x i8] c";unknown;unknown;0;0;;\00", align 1
12@0 = private unnamed_addr global %struct.ident_t { i32 0, i32 2, i32 0, i32 0, ptr @.str }, align 8
13
14; CHECK-LABEL: {{[^@]+}}Successfully got offload values:
15; CHECK-NEXT: offload_baseptrs: ptr %a ---   %size.addr = alloca i32, align 4 ---
16; CHECK-NEXT: offload_ptrs: ptr %a ---   %size.addr = alloca i32, align 4 ---
17; CHECK-NEXT: offload_sizes:   %0 = shl nuw nsw i64 %conv, 3 --- i64 4 ---
18
19;int heavyComputation(ptr a, unsigned size) {
20;  int random = rand() % 7;
21;
22;  //#pragma omp target data map(a[0:size], size)
23;  ptr args[2];
24;  args[0] = &a;
25;  args[1] = &size;
26;  __tgt_target_data_begin(..., args, ...)
27;
28;  #pragma omp target teams
29;  for (int i = 0; i < size; ++i) {
30;    a[i] = ++aptr 3.141624;
31;  }
32;
33;  return random;
34;}
35define dso_local i32 @heavyComputation(ptr %a, i32 %size) {
36entry:
37  %size.addr = alloca i32, align 4
38  %.offload_baseptrs = alloca [2 x ptr], align 8
39  %.offload_ptrs = alloca [2 x ptr], align 8
40  %.offload_sizes = alloca [2 x i64], align 8
41
42  store i32 %size, ptr %size.addr, align 4
43  %call = tail call i32 (...) @rand()
44
45  %conv = zext i32 %size to i64
46  %0 = shl nuw nsw i64 %conv, 3
47  store ptr %a, ptr %.offload_baseptrs, align 8
48  store ptr %a, ptr %.offload_ptrs, align 8
49  store i64 %0, ptr %.offload_sizes, align 8
50  %1 = getelementptr inbounds [2 x ptr], ptr %.offload_baseptrs, i64 0, i64 1
51  store ptr %size.addr, ptr %1, align 8
52  %2 = getelementptr inbounds [2 x ptr], ptr %.offload_ptrs, i64 0, i64 1
53  store ptr %size.addr, ptr %2, align 8
54  %3 = getelementptr inbounds [2 x i64], ptr %.offload_sizes, i64 0, i64 1
55  store i64 4, ptr %3, align 8
56  call void @__tgt_target_data_begin_mapper(ptr @0, i64 -1, i32 2, ptr nonnull %.offload_baseptrs, ptr nonnull %.offload_ptrs, ptr nonnull %.offload_sizes, ptr @.offload_maptypes., ptr null, ptr null)
57  %rem = srem i32 %call, 7
58  call void @__tgt_target_data_end_mapper(ptr @0, i64 -1, i32 2, ptr nonnull %.offload_baseptrs, ptr nonnull %.offload_ptrs, ptr nonnull %.offload_sizes, ptr @.offload_maptypes., ptr null, ptr null)
59  ret i32 %rem
60}
61
62declare void @__tgt_target_data_begin_mapper(ptr, i64, i32, ptr, ptr, ptr, ptr, ptr, ptr)
63declare void @__tgt_target_data_end_mapper(ptr, i64, i32, ptr, ptr, ptr, ptr, ptr, ptr)
64
65declare dso_local i32 @rand(...)
66
67!llvm.module.flags = !{!0}
68
69!0 = !{i32 7, !"openmp", i32 50}
70
71