mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-25 00:36:06 +00:00
f9734b9df1
This PR adds support to translate the `private` clause from MLIR to LLVMIR when used on allocatables in the context of an `omp.target` op. This replaces https://github.com/llvm/llvm-project/pull/113208. Parent PR: https://github.com/llvm/llvm-project/pull/116770. Only the latest commit is relevant to the PR.
42 lines
2.1 KiB
MLIR
42 lines
2.1 KiB
MLIR
// RUN: mlir-translate -mlir-to-llvmir %s | FileCheck %s
|
|
|
|
// This tests check that target code nested inside a target data region which
|
|
// has only use_device_ptr mapping corectly generates code on the device pass.
|
|
|
|
// CHECK: define weak_odr protected void @__omp_offloading{{.*}}main_
|
|
// CHECK-NEXT: entry:
|
|
// CHECK-NEXT: %[[VAL_3:.*]] = alloca ptr, align 8
|
|
// CHECK-NEXT: store ptr %[[VAL_4:.*]], ptr %[[VAL_3]], align 8
|
|
// CHECK-NEXT: %[[VAL_5:.*]] = call i32 @__kmpc_target_init(ptr @__omp_offloading_{{.*}}_kernel_environment, ptr %[[VAL_6:.*]])
|
|
// CHECK-NEXT: %[[VAL_7:.*]] = icmp eq i32 %[[VAL_5]], -1
|
|
// CHECK-NEXT: br i1 %[[VAL_7]], label %[[VAL_8:.*]], label %[[VAL_9:.*]]
|
|
// CHECK: user_code.entry: ; preds = %[[VAL_10:.*]]
|
|
// CHECK-NEXT: %[[VAL_11:.*]] = load ptr, ptr %[[VAL_3]], align 8
|
|
// CHECK-NEXT: br label %[[VAL_12:.*]]
|
|
|
|
// CHECK: [[VAL_12]]:
|
|
// CHECK-NEXT: br label %[[TARGET_REG_ENTRY:.*]]
|
|
|
|
// CHECK: [[TARGET_REG_ENTRY]]: ; preds = %[[VAL_12]]
|
|
// CHECK-NEXT: %[[VAL_13:.*]] = load ptr, ptr %[[VAL_11]], align 8
|
|
// CHECK-NEXT: store i32 999, ptr %[[VAL_13]], align 4
|
|
// CHECK-NEXT: br label %[[VAL_14:.*]]
|
|
module attributes {omp.is_target_device = true } {
|
|
llvm.func @_QQmain() attributes {fir.bindc_name = "main"} {
|
|
%0 = llvm.mlir.constant(1 : i64) : i64
|
|
%a = llvm.alloca %0 x !llvm.ptr : (i64) -> !llvm.ptr
|
|
%map = omp.map.info var_ptr(%a : !llvm.ptr, !llvm.ptr) map_clauses(tofrom) capture(ByRef) -> !llvm.ptr {name = ""}
|
|
omp.target_data use_device_ptr(%map -> %arg0 : !llvm.ptr) {
|
|
%map1 = omp.map.info var_ptr(%arg0 : !llvm.ptr, !llvm.ptr) map_clauses(tofrom) capture(ByRef) -> !llvm.ptr {name = ""}
|
|
omp.target map_entries(%map1 -> %arg1 : !llvm.ptr){
|
|
%1 = llvm.mlir.constant(999 : i32) : i32
|
|
%2 = llvm.load %arg1 : !llvm.ptr -> !llvm.ptr
|
|
llvm.store %1, %2 : i32, !llvm.ptr
|
|
omp.terminator
|
|
}
|
|
omp.terminator
|
|
}
|
|
llvm.return
|
|
}
|
|
}
|