| // RUN: mlir-translate -mlir-to-llvmir %s | FileCheck %s |
| |
| omp.private {type = private} @privatizer : i32 |
| |
| omp.private {type = firstprivate} @firstprivatizer : i32 copy { |
| ^bb0(%arg0: !llvm.ptr, %arg1: !llvm.ptr): |
| %0 = llvm.load %arg0 : !llvm.ptr -> i32 |
| llvm.store %0, %arg1 : i32, !llvm.ptr |
| omp.yield(%arg1 : !llvm.ptr) |
| } |
| |
| llvm.func @task_privatization_test() { |
| %c0 = llvm.mlir.constant(0: i32) : i32 |
| %c1 = llvm.mlir.constant(1: i32) : i32 |
| %0 = llvm.alloca %c1 x i32 : (i32) -> !llvm.ptr |
| %1 = llvm.alloca %c1 x i32 : (i32) -> !llvm.ptr |
| llvm.store %c0, %0 : i32, !llvm.ptr |
| llvm.store %c1, %1 : i32, !llvm.ptr |
| |
| omp.task private(@privatizer %0 -> %arg0, @firstprivatizer %1 -> %arg1 : !llvm.ptr, !llvm.ptr) { |
| %2 = llvm.load %arg1 : !llvm.ptr -> i32 |
| llvm.store %2, %arg0 : i32, !llvm.ptr |
| omp.terminator |
| } |
| llvm.return |
| } |
| |
| // CHECK: define void @task_privatization_test() |
| // CHECK: %[[STRUCT_ARG:.*]] = alloca { ptr }, align 8 |
| // CHECK: %[[VAL_0:.*]] = alloca i32, align 4 |
| // CHECK: %[[VAL_1:.*]] = alloca i32, align 4 |
| // CHECK: store i32 0, ptr %[[VAL_0]], align 4 |
| // CHECK: store i32 1, ptr %[[VAL_1]], align 4 |
| // CHECK: br label %entry |
| // CHECK: entry: |
| // CHECK: br label %omp.private.init |
| // CHECK: omp.private.init: |
| // CHECK: %[[VAL_5:.*]] = tail call ptr @malloc(i64 ptrtoint (ptr getelementptr ([[STRUCT_KMP_PRIVATES_T:.*]], ptr null, i32 1) to i64)) |
| // CHECK: %[[VAL_7:.*]] = getelementptr { i32 }, ptr %[[VAL_5]], i32 0, i32 0 |
| // CHECK: br label %omp.private.copy |
| // CHECK: omp.private.copy: |
| // CHECK: %[[VAL_10:.*]] = load i32, ptr %[[VAL_1]], align 4 |
| // CHECK: store i32 %[[VAL_10]], ptr %[[VAL_7]], align 4 |
| // CHECK: br label %omp.task.start |
| // CHECK: omp.task.start: |
| // CHECK: br label %codeRepl |
| // CHECK: codeRepl: |
| // CHECK: %[[GEP_OMP_TASK_CONTEXT_PTR:.*]] = getelementptr { ptr }, ptr %[[STRUCT_ARG]], i32 0, i32 0 |
| // CHECK: store ptr %[[VAL_5]], ptr %[[GEP_OMP_TASK_CONTEXT_PTR]], align 8 |
| // CHECK: %[[VAL_14:.*]] = call i32 @__kmpc_global_thread_num(ptr @1) |
| // 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) |
| // CHECK: %[[ALLOCATED_TASK_STRUCT:.*]] = load ptr, ptr %[[VAL_15]], align 8 |
| // CHECK: call void @llvm.memcpy.p0.p0.i64(ptr align 1 %[[ALLOCATED_TASK_STRUCT]], ptr align 1 %[[STRUCT_ARG]], i64 8, i1 false) |
| // CHECK: %[[VAL_16:.*]] = call i32 @__kmpc_omp_task(ptr @1, i32 %[[VAL_14]], ptr %[[VAL_15]]) |
| // CHECK: br label %[[VAL_17:.*]] |
| // CHECK: task.exit: |
| // CHECK: ret void |
| |
| // CHECK-LABEL: define internal void @task_privatization_test..omp_par( |
| // CHECK-SAME: i32 %[[GLOBAL_TID_VAL:.*]], ptr %[[OMP_TASK_CONTEXT_PTR_PTR_PTR_PTR:.*]]) |
| // CHECK: task.alloca: |
| // CHECK: %[[OMP_TASK_CONEXT_PTR_PTR_PTR:.*]] = load ptr, ptr %[[OMP_TASK_CONTEXT_PTR_PTR_PTR_PTR]], align 8 |
| // CHECK: %[[OMP_TASK_CONTEXT_PTR_PTR:.*]] = getelementptr { ptr }, ptr %[[OMP_TASK_CONTEXT_PTR_PTR_PTR:.*]], i32 0, i32 0 |
| // CHECK: %[[OMP_TASK_CONTEXT_PTR:.*]] = load ptr, ptr %[[OMP_TASK_CONTEXT_PTR_PTR:.*]], align 8 |
| // CHECK: %[[OMP_PRIVATE_ALLOC:.*]] = alloca i32, align 4 |
| // CHECK: br label %[[VAL_18:.*]] |
| // CHECK: task.body: ; preds = %[[VAL_19:.*]] |
| // CHECK: %[[VAL_20:.*]] = getelementptr { i32 }, ptr %[[OMP_TASK_CONTEXT_PTR]], i32 0, i32 0 |
| // CHECK: br label %[[VAL_23:.*]] |
| // CHECK: omp.task.region: ; preds = %[[VAL_18]] |
| // CHECK: %[[VAL_24:.*]] = load i32, ptr %[[VAL_20]], align 4 |
| // CHECK: store i32 %[[VAL_24]], ptr %[[OMP_PRIVATE_ALLOC]], align 4 |
| // CHECK: br label %[[VAL_25:.*]] |
| // CHECK: omp.region.cont: ; preds = %[[VAL_23]] |
| // CHECK: tail call void @free(ptr %[[OMP_TASK_CONTEXT_PTR]]) |
| // CHECK: br label %[[VAL_26:.*]] |
| // CHECK: task.exit.exitStub: ; preds = %[[VAL_25]] |
| // CHECK: ret void |
| |
