80 lines · plain
1// RUN: mlir-translate -mlir-to-llvmir %s | FileCheck %s2 3omp.private {type = private} @privatizer : i324 5omp.private {type = firstprivate} @firstprivatizer : i32 copy {6^bb0(%arg0: !llvm.ptr, %arg1: !llvm.ptr):7 %0 = llvm.load %arg0 : !llvm.ptr -> i328 llvm.store %0, %arg1 : i32, !llvm.ptr9 omp.yield(%arg1 : !llvm.ptr)10}11 12llvm.func @task_privatization_test() {13 %c0 = llvm.mlir.constant(0: i32) : i3214 %c1 = llvm.mlir.constant(1: i32) : i3215 %0 = llvm.alloca %c1 x i32 : (i32) -> !llvm.ptr16 %1 = llvm.alloca %c1 x i32 : (i32) -> !llvm.ptr17 llvm.store %c0, %0 : i32, !llvm.ptr18 llvm.store %c1, %1 : i32, !llvm.ptr19 20 omp.task private(@privatizer %0 -> %arg0, @firstprivatizer %1 -> %arg1 : !llvm.ptr, !llvm.ptr) {21 %2 = llvm.load %arg1 : !llvm.ptr -> i3222 llvm.store %2, %arg0 : i32, !llvm.ptr23 omp.terminator24 }25 llvm.return26}27 28// CHECK: define void @task_privatization_test()29// CHECK: %[[STRUCT_ARG:.*]] = alloca { ptr }, align 830// CHECK: %[[VAL_0:.*]] = alloca i32, align 431// CHECK: %[[VAL_1:.*]] = alloca i32, align 432// CHECK: store i32 0, ptr %[[VAL_0]], align 433// CHECK: store i32 1, ptr %[[VAL_1]], align 434// CHECK: br label %entry35// CHECK: entry:36// CHECK: br label %omp.private.init37// CHECK: omp.private.init:38// CHECK: %[[VAL_5:.*]] = tail call ptr @malloc(i64 ptrtoint (ptr getelementptr ([[STRUCT_KMP_PRIVATES_T:.*]], ptr null, i32 1) to i64))39// CHECK: %[[VAL_7:.*]] = getelementptr { i32 }, ptr %[[VAL_5]], i32 0, i32 040// CHECK: br label %omp.private.copy41// CHECK: omp.private.copy:42// CHECK: %[[VAL_10:.*]] = load i32, ptr %[[VAL_1]], align 443// CHECK: store i32 %[[VAL_10]], ptr %[[VAL_7]], align 444// CHECK: br label %omp.task.start45// CHECK: omp.task.start:46// CHECK: br label %codeRepl47// CHECK: codeRepl:48// CHECK: %[[GEP_OMP_TASK_CONTEXT_PTR:.*]] = getelementptr { ptr }, ptr %[[STRUCT_ARG]], i32 0, i32 049// CHECK: store ptr %[[VAL_5]], ptr %[[GEP_OMP_TASK_CONTEXT_PTR]], align 850// CHECK: %[[VAL_14:.*]] = call i32 @__kmpc_global_thread_num(ptr @1)51// CHECK: %[[VAL_15:.*]] = call ptr @__kmpc_omp_task_alloc(ptr @1, i32 %[[VAL_14]], i32 1, i64 40, i64 8, ptr @task_privatization_test..omp_par)52// CHECK: %[[ALLOCATED_TASK_STRUCT:.*]] = load ptr, ptr %[[VAL_15]], align 853// CHECK: call void @llvm.memcpy.p0.p0.i64(ptr align 1 %[[ALLOCATED_TASK_STRUCT]], ptr align 1 %[[STRUCT_ARG]], i64 8, i1 false)54// CHECK: %[[VAL_16:.*]] = call i32 @__kmpc_omp_task(ptr @1, i32 %[[VAL_14]], ptr %[[VAL_15]])55// CHECK: br label %[[VAL_17:.*]]56// CHECK: task.exit:57// CHECK: ret void58 59// CHECK-LABEL: define internal void @task_privatization_test..omp_par(60// CHECK-SAME: i32 %[[GLOBAL_TID_VAL:.*]], ptr %[[OMP_TASK_CONTEXT_PTR_PTR_PTR_PTR:.*]])61// CHECK: task.alloca:62// CHECK: %[[OMP_TASK_CONEXT_PTR_PTR_PTR:.*]] = load ptr, ptr %[[OMP_TASK_CONTEXT_PTR_PTR_PTR_PTR]], align 863// CHECK: %[[OMP_TASK_CONTEXT_PTR_PTR:.*]] = getelementptr { ptr }, ptr %[[OMP_TASK_CONTEXT_PTR_PTR_PTR:.*]], i32 0, i32 064// CHECK: %[[OMP_TASK_CONTEXT_PTR:.*]] = load ptr, ptr %[[OMP_TASK_CONTEXT_PTR_PTR:.*]], align 865// CHECK: %[[OMP_PRIVATE_ALLOC:.*]] = alloca i32, align 466// CHECK: br label %[[VAL_18:.*]]67// CHECK: task.body: ; preds = %[[VAL_19:.*]]68// CHECK: %[[VAL_20:.*]] = getelementptr { i32 }, ptr %[[OMP_TASK_CONTEXT_PTR]], i32 0, i32 069// CHECK: br label %[[VAL_23:.*]]70// CHECK: omp.task.region: ; preds = %[[VAL_18]]71// CHECK: %[[VAL_24:.*]] = load i32, ptr %[[VAL_20]], align 472// CHECK: store i32 %[[VAL_24]], ptr %[[OMP_PRIVATE_ALLOC]], align 473// CHECK: br label %[[VAL_25:.*]]74// CHECK: omp.region.cont: ; preds = %[[VAL_23]]75// CHECK: tail call void @free(ptr %[[OMP_TASK_CONTEXT_PTR]])76// CHECK: br label %[[VAL_26:.*]]77// CHECK: task.exit.exitStub: ; preds = %[[VAL_25]]78// CHECK: ret void79 80