mlir/test/Target/LLVMIR/openmp-nested.mlir

   1 // RUN: mlir-translate -mlir-to-llvmir -split-input-file %s | FileCheck %s
   2
   3 module {
   4   llvm.func @printf(!llvm.ptr<i8>, ...) -> i32
   5   llvm.mlir.global internal constant @str0("WG size of kernel = %d X %d\0A\00")
   6
   7   llvm.func @main(%arg0: i32, %arg1: !llvm.ptr<ptr<i8>>) -> i32 {
   8     omp.parallel   {
   9       %0 = llvm.mlir.constant(1 : index) : i64
  10       %1 = llvm.mlir.constant(10 : index) : i64
  11       %2 = llvm.mlir.constant(0 : index) : i64
  12       %4 = llvm.mlir.constant(0 : i32) : i32
  13       %12 = llvm.alloca %0 x i64 : (i64) -> !llvm.ptr<i64>
  14       omp.wsloop for (%arg2) : i64 = (%2) to (%1) step (%0)  {
  15         omp.parallel   {
  16           omp.wsloop for (%arg3) : i64 = (%2) to (%0) step (%0)  {
  17             llvm.store %2, %12 : !llvm.ptr<i64>
  18             omp.yield
  19           }
  20           omp.terminator
  21         }
  22         %19 = llvm.load %12 : !llvm.ptr<i64>
  23         %20 = llvm.trunc %19 : i64 to i32
  24         %5 = llvm.mlir.addressof @str0 : !llvm.ptr<array<29 x i8>>
  25         %6 = llvm.getelementptr %5[%4, %4] : (!llvm.ptr<array<29 x i8>>, i32, i32) -> !llvm.ptr<i8>
  26         %21 = llvm.call @printf(%6, %20, %20) : (!llvm.ptr<i8>, i32, i32) -> i32
  27         omp.yield
  28       }
  29       omp.terminator
  30     }
  31     %a4 = llvm.mlir.constant(0 : i32) : i32
  32     llvm.return %a4 : i32
  33   }
  34
  35 }
  36
  37 // CHECK: call void (ptr, i32, ptr, ...) @__kmpc_fork_call(ptr @1, i32 0, ptr @[[inner1:.+]])
  38
  39 // CHECK: define internal void @[[inner1]]
  40 // CHECK: %[[structArg:.+]] = alloca { ptr }
  41 // CHECK: call void (ptr, i32, ptr, ...) @__kmpc_fork_call(ptr @1, i32 1, ptr @[[inner2:.+]], ptr %[[structArg]])