102 lines · plain
1// RUN: mlir-translate -mlir-to-llvmir -split-input-file %s 2>&1 | FileCheck %s2 3module attributes {omp.target_triples = ["amdgcn-amd-amdhsa"]} {4 llvm.func @_QPopenmp_target_data_enter() {5 %0 = llvm.mlir.constant(1 : i64) : i646 %1 = llvm.alloca %0 x i32 {bindc_name = "i", in_type = i32, operand_segment_sizes = array<i32: 0, 0>, uniq_name = "_QFopenmp_target_dataEi"} : (i64) -> !llvm.ptr7 %2 = omp.map.info var_ptr(%1 : !llvm.ptr, i32) map_clauses(to) capture(ByRef) -> !llvm.ptr {name = ""}8 9 omp.target_enter_data map_entries(%2 : !llvm.ptr) nowait10 11 llvm.return12 }13}14 15// CHECK: define void @_QPopenmp_target_data_enter() {16 17// CHECK: %[[TASK:.*]] = call ptr @__kmpc_omp_target_task_alloc18// CHECK-SAME: (ptr @{{.*}}, i32 %{{.*}}, i32 {{.*}}, i64 {{.*}}, i64 {{.*}}, ptr19// CHECK-SAME: @[[TASK_PROXY_FUNC_ENTER:.*]], i64 {{.*}})20 21// CHECK: call i32 @__kmpc_omp_task(ptr {{.*}}, i32 %{{.*}}, ptr %[[TASK]])22// CHECK: }23 24// CHECK: define internal void @[[TASK_BODY_FUNC_ENTER:.*]](i32 {{.*}}, ptr %[[OFFLOAD_BASE_PTRS:.*]], ptr %[[OFFLOAD_PTRS:.*]]) {25// CHECK: call void @__tgt_target_data_begin_nowait_mapper(26// CHECK-SAME: ptr @{{.*}}, i64 -1, i32 1,27// CHECK-SAME: ptr %[[OFFLOAD_BASE_PTRS]], ptr %[[OFFLOAD_PTRS]],28// CHECK-SAME: ptr @{{.*}}, ptr @{{.*}}, ptr @{{.*}}, ptr null, i32 0, ptr null, i32 0, ptr null)29// CHECK: }30 31// CHECK: define internal void @[[TASK_PROXY_FUNC_ENTER]](i32 %{{.*}}, ptr %{{.*}}) {32// CHECK: call void @[[TASK_BODY_FUNC_ENTER]](i32 %{{.*}}, ptr %{{.*}}, ptr %{{.*}})33// CHECK: }34 35// -----36 37module attributes {omp.target_triples = ["amdgcn-amd-amdhsa"]} {38 llvm.func @_QPopenmp_target_data_update() {39 %0 = llvm.mlir.constant(1 : i64) : i6440 %1 = llvm.alloca %0 x i32 {bindc_name = "i", in_type = i32, operand_segment_sizes = array<i32: 0, 0>, uniq_name = "_QFopenmp_target_dataEi"} : (i64) -> !llvm.ptr41 %2 = omp.map.info var_ptr(%1 : !llvm.ptr, i32) map_clauses(to) capture(ByRef) -> !llvm.ptr {name = ""}42 43 omp.target_update map_entries(%2 : !llvm.ptr) nowait44 45 llvm.return46 }47}48 49// CHECK: define void @_QPopenmp_target_data_update() {50 51// CHECK: %[[TASK:.*]] = call ptr @__kmpc_omp_target_task_alloc52// CHECK-SAME: (ptr @{{.*}}, i32 %{{.*}}, i32 {{.*}}, i64 {{.*}}, i64 {{.*}}, ptr53// CHECK-SAME: @[[TASK_PROXY_FUNC_UPDATE:.*]], i64 {{.*}})54 55// CHECK: call i32 @__kmpc_omp_task(ptr {{.*}}, i32 %{{.*}}, ptr %[[TASK]])56// CHECK: }57 58// CHECK: define internal void @[[TASK_BODY_FUNC_UPDATE:.*]](i32 {{.*}}, ptr %[[OFFLOAD_BASE_PTRS:.*]], ptr %[[OFFLOAD_PTRS:.*]]) {59// CHECK: call void @__tgt_target_data_update_nowait_mapper(60// CHECK-SAME: ptr @{{.*}}, i64 -1, i32 1,61// CHECK-SAME: ptr %[[OFFLOAD_BASE_PTRS]], ptr %[[OFFLOAD_PTRS]],62// CHECK-SAME: ptr @{{.*}}, ptr @{{.*}}, ptr @{{.*}}, ptr null, i32 0, ptr null, i32 0, ptr null)63// CHECK: }64 65// CHECK: define internal void @[[TASK_PROXY_FUNC_UPDATE]](i32 %{{.*}}, ptr %{{.*}}) {66// CHECK: call void @[[TASK_BODY_FUNC_UPDATE]](i32 %{{.*}}, ptr %{{.*}})67// CHECK: }68 69// -----70 71module attributes {omp.target_triples = ["amdgcn-amd-amdhsa"]} {72 llvm.func @_QPopenmp_target_data_exit() {73 %0 = llvm.mlir.constant(1 : i64) : i6474 %1 = llvm.alloca %0 x i32 {bindc_name = "i", in_type = i32, operand_segment_sizes = array<i32: 0, 0>, uniq_name = "_QFopenmp_target_dataEi"} : (i64) -> !llvm.ptr75 %2 = omp.map.info var_ptr(%1 : !llvm.ptr, i32) map_clauses(from) capture(ByRef) -> !llvm.ptr {name = ""}76 77 omp.target_exit_data map_entries(%2 : !llvm.ptr) nowait78 79 llvm.return80 }81}82 83// CHECK: define void @_QPopenmp_target_data_exit() {84 85// CHECK: %[[TASK:.*]] = call ptr @__kmpc_omp_target_task_alloc86// CHECK-SAME: (ptr @{{.*}}, i32 %{{.*}}, i32 {{.*}}, i64 {{.*}}, i64 {{.*}}, ptr87// CHECK-SAME: @[[TASK_PROXY_FUNC_EXIT:.*]], i64 {{.*}})88 89// CHECK: call i32 @__kmpc_omp_task(ptr {{.*}}, i32 %{{.*}}, ptr %[[TASK]])90// CHECK: }91 92// CHECK: define internal void @[[TASK_BODY_FUNC_EXIT:.*]](i32 %{{.*}}, ptr %[[OFFLOAD_BASE_PTRS:.*]], ptr %[[OFFLOAD_PTRS:.*]]) {93// CHECK: call void @__tgt_target_data_end_nowait_mapper(94// CHECK-SAME: ptr @{{.*}}, i64 -1, i32 1,95// CHECK-SAME: ptr %[[OFFLOAD_BASE_PTRS]], ptr %[[OFFLOAD_PTRS]],96// CHECK-SAME: ptr @{{.*}}, ptr @{{.*}}, ptr @{{.*}}, ptr null, i32 0, ptr null, i32 0, ptr null)97// CHECK: }98 99// CHECK: define internal void @[[TASK_PROXY_FUNC_EXIT]](i32 %{{.*}}, ptr %{{.*}}) {100// CHECK: call void @[[TASK_BODY_FUNC_EXIT]](i32 %{{.*}}, ptr %{{.*}})101// CHECK: }102