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, ...) -> 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) -> 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
  14       omp.wsloop {
  15         omp.loop_nest (%arg2) : i64 = (%2) to (%1) step (%0) {
  16           omp.parallel {
  17             omp.wsloop {
  18               omp.loop_nest (%arg3) : i64 = (%2) to (%0) step (%0) {
  19                 llvm.store %2, %12 : i64, !llvm.ptr
  20                 omp.yield
  21               }
  22             }
  23             omp.terminator
  24           }
  25           %19 = llvm.load %12 : !llvm.ptr -> i64
  26           %20 = llvm.trunc %19 : i64 to i32
  27           %5 = llvm.mlir.addressof @str0 : !llvm.ptr
  28           %6 = llvm.getelementptr %5[%4, %4] : (!llvm.ptr, i32, i32) -> !llvm.ptr, !llvm.array<29 x i8>
  29           %21 = llvm.call @printf(%6, %20, %20) vararg(!llvm.func<i32 (ptr, ...)>): (!llvm.ptr, i32, i32) -> i32
  30           omp.yield
  31         }
  32       }
  33       omp.terminator
  34     }
  35     %a4 = llvm.mlir.constant(0 : i32) : i32
  36     llvm.return %a4 : i32
  37   }
  38
  39 }
  40
  41 // CHECK: call void (ptr, i32, ptr, ...) @__kmpc_fork_call(ptr @1, i32 0, ptr @[[inner1:.+]])
  42
  43 // CHECK: define internal void @[[inner1]]
  44 // CHECK: %[[structArg:.+]] = alloca { ptr }
  45 // CHECK: call void (ptr, i32, ptr, ...) @__kmpc_fork_call(ptr @1, i32 1, ptr @[[inner2:.+]], ptr %[[structArg]])