mlir/test/Target/LLVMIR/omptarget-depend-host-only.mlir

   1 // RUN: mlir-translate -mlir-to-llvmir %s | FileCheck %s
   2
   3 module attributes {omp.is_target_device = false} {
   4   llvm.func @omp_target_depend_() {
   5     %0 = llvm.mlir.constant(39 : index) : i64
   6     %1 = llvm.mlir.constant(1 : index) : i64
   7     %2 = llvm.mlir.constant(40 : index) : i64
   8     %3 = omp.map.bounds lower_bound(%1 : i64) upper_bound(%0 : i64) extent(%2 : i64) stride(%1 : i64) start_idx(%1 : i64)
   9     %4 = llvm.mlir.addressof @_QFEa : !llvm.ptr
  10     %5 = omp.map.info var_ptr(%4 : !llvm.ptr, !llvm.array<40 x i32>) map_clauses(from) capture(ByRef) bounds(%3) -> !llvm.ptr {name = "a"}
  11     omp.target depend(taskdependin -> %4 : !llvm.ptr) map_entries(%5 -> %arg0 : !llvm.ptr) {
  12       %6 = llvm.mlir.constant(100 : index) : i32
  13       llvm.store %6, %arg0 : i32, !llvm.ptr
  14       omp.terminator
  15     }
  16     llvm.return
  17   }
  18
  19   llvm.mlir.global internal @_QFEa() {addr_space = 0 : i32} : !llvm.array<40 x i32> {
  20     %0 = llvm.mlir.zero : !llvm.array<40 x i32>
  21     llvm.return %0 : !llvm.array<40 x i32>
  22   }
  23 }
  24
  25 // CHECK: define void @omp_target_depend_()
  26 // CHECK-NOT: define {{.*}} @
  27 // CHECK-NOT: call i32 @__tgt_target_kernel({{.*}})
  28 // CHECK: call void @__omp_offloading_[[DEV:.*]]_[[FIL:.*]]_omp_target_depend__l[[LINE:.*]](ptr {{.*}})
  29 // CHECK-NEXT: ret void
  30
  31 // CHECK: define internal void @__omp_offloading_[[DEV]]_[[FIL]]_omp_target_depend__l[[LINE]](ptr %[[ADDR_A:.*]])
  32 // CHECK: store i32 100, ptr %[[ADDR_A]], align 4