47 lines · plain
1// RUN: mlir-translate -mlir-to-llvmir %s | FileCheck %s2 3module attributes {omp.is_target_device = false, omp.target_triples = ["amdgcn-amd-amdhsa"]} {4 llvm.func @target_if_nowait(%arg0: !llvm.ptr, %arg1: !llvm.ptr) {5 %0 = llvm.mlir.constant(1 : i64) : i646 %3 = llvm.alloca %0 x i32 {bindc_name = "cond"} : (i64) -> !llvm.ptr7 %6 = llvm.load %3 : !llvm.ptr -> i328 %7 = llvm.mlir.constant(0 : i64) : i329 %8 = llvm.icmp "ne" %6, %7 : i3210 %9 = omp.map.info var_ptr(%3 : !llvm.ptr, i32) map_clauses(implicit, exit_release_or_enter_alloc) capture(ByCopy) -> !llvm.ptr {name = "cond"}11 %10 = omp.map.info var_ptr(%arg0 : !llvm.ptr, f32) map_clauses(implicit, exit_release_or_enter_alloc) capture(ByCopy) -> !llvm.ptr {name = "var"}12 %11 = omp.map.info var_ptr(%arg1 : !llvm.ptr, f32) map_clauses(implicit, exit_release_or_enter_alloc) capture(ByCopy) -> !llvm.ptr {name = "val"}13 omp.target if(%8) nowait map_entries(%10 -> %arg3, %11 -> %arg4 : !llvm.ptr, !llvm.ptr) {14 %12 = llvm.load %arg4 : !llvm.ptr -> f3215 llvm.store %12, %arg3 : f32, !llvm.ptr16 omp.terminator17 }18 llvm.return19 }20}21 22// CHECK: define void @target_if_nowait{{.*}} {23// CHECK: omp_if.then:24// CHECK: br label %[[TARGET_TASK_BB:.*]]25 26// CHECK: [[TARGET_TASK_BB]]:27// CHECK: call ptr @__kmpc_omp_target_task_alloc28// CHECK: br label %[[OFFLOAD_CONT:.*]]29 30// CHECK: [[OFFLOAD_CONT]]:31// CHECK: br label %omp_if.end32 33// CHECK: omp_if.else:34// CHECK: br label %[[HOST_TASK_BB:.*]]35 36// CHECK: [[HOST_TASK_BB]]:37// CHECK: call ptr @__kmpc_omp_task_alloc38// CHECK: br label %[[HOST_TASK_CONT:.*]]39 40// CHECK: [[HOST_TASK_CONT]]:41// CHECK: br label %omp_if.end42 43// CHECK: omp_if.end:44// CHECK: ret void45// CHECK: }46 47