mlir/test/Dialect/OpenMP/cli-canonical_loop.mlir

// RUN: mlir-opt %s            | FileCheck %s --enable-var-scope
// RUN: mlir-opt %s | mlir-opt | FileCheck %s --enable-var-scope


// CHECK-LABEL: @omp_canonloop_raw(
// CHECK-SAME: %[[tc:.+]]: i32)
func.func @omp_canonloop_raw(%tc : i32) -> () {
  // CHECK: omp.canonical_loop %iv : i32 in range(%[[tc]]) {
  "omp.canonical_loop" (%tc) ({
    ^bb0(%iv: i32):
      // CHECK-NEXT: = llvm.add %iv, %iv : i32
      %newval = llvm.add %iv, %iv : i32
      // CHECK-NEXT: omp.terminator
      omp.terminator
  // CHECK-NEXT: }
  }) : (i32) -> ()
  // CHECK-NEXT: return
  return
}


// CHECK-LABEL: @omp_canonloop_sequential_raw(
// CHECK-SAME: %[[tc:.+]]: i32)
func.func @omp_canonloop_sequential_raw(%tc : i32) -> () {
  // CHECK-NEXT: %canonloop_s0 = omp.new_cli
  %canonloop_s0 = "omp.new_cli" () : () -> (!omp.cli)
  // CHECK-NEXT:  omp.canonical_loop(%canonloop_s0) %iv_s0 : i32 in range(%[[tc]]) {
  "omp.canonical_loop" (%tc, %canonloop_s0) ({
    ^bb_first(%iv_first: i32):
      // CHECK-NEXT: = llvm.add %iv_s0, %iv_s0 : i32
      %newval = llvm.add %iv_first, %iv_first : i32
    // CHECK-NEXT: omp.terminator
    omp.terminator
  // CHECK-NEXT: }
  }) : (i32, !omp.cli) -> ()

  // CHECK-NEXT: %canonloop_s1 = omp.new_cli
  %canonloop_s1 = "omp.new_cli" () : () -> (!omp.cli)
  // CHECK-NEXT: omp.canonical_loop(%canonloop_s1) %iv_s1 : i32 in range(%[[tc]]) {
  "omp.canonical_loop" (%tc, %canonloop_s1) ({
    ^bb_second(%iv_second: i32):
    // CHECK: omp.terminator
    omp.terminator
  // CHECK-NEXT: }
  }) : (i32, !omp.cli) -> ()

  // CHECK-NEXT: return
  return
}


// CHECK-LABEL: @omp_nested_canonloop_raw(
// CHECK-SAME: %[[tc_outer:.+]]: i32, %[[tc_inner:.+]]: i32)
func.func @omp_nested_canonloop_raw(%tc_outer : i32, %tc_inner : i32) -> () {
  // CHECK-NEXT: %canonloop = omp.new_cli
  %outer = "omp.new_cli" () : () -> (!omp.cli)
  // CHECK-NEXT: %canonloop_d1 = omp.new_cli
  %inner = "omp.new_cli" () : () -> (!omp.cli)
  // CHECK-NEXT: omp.canonical_loop(%canonloop) %iv : i32 in range(%[[tc_outer]]) {
  "omp.canonical_loop" (%tc_outer, %outer) ({
    ^bb_outer(%iv_outer: i32):
      // CHECK-NEXT: omp.canonical_loop(%canonloop_d1) %iv_d1 : i32 in range(%[[tc_inner]]) {
      "omp.canonical_loop" (%tc_inner, %inner) ({
        ^bb_inner(%iv_inner: i32):
          // CHECK-NEXT: = llvm.add %iv, %iv_d1 : i32
          %newval = llvm.add %iv_outer, %iv_inner: i32
          // CHECK-NEXT: omp.terminator
          omp.terminator
      }) : (i32, !omp.cli) -> ()
      // CHECK: omp.terminator
      omp.terminator
  }) : (i32, !omp.cli) -> ()
  return
}


// CHECK-LABEL: @omp_canonloop_pretty(
// CHECK-SAME: %[[tc:.+]]: i32)
func.func @omp_canonloop_pretty(%tc : i32) -> () {
  // CHECK-NEXT: omp.canonical_loop %iv : i32 in range(%[[tc]]) {
  omp.canonical_loop %iv : i32 in range(%tc) {
    // CHECK-NEXT: llvm.add %iv, %iv : i32
    %newval = llvm.add %iv, %iv: i32
    // CHECK-NEXT: omp.terminator
    omp.terminator
  }
  return
}


// CHECK-LABEL: @omp_canonloop_constant_pretty()
func.func @omp_canonloop_constant_pretty() -> () {
  // CHECK-NEXT:  %[[tc:.+]] = llvm.mlir.constant(4 : i32) : i32
  %tc = llvm.mlir.constant(4 : i32) : i32
  // CHECK-NEXT: omp.canonical_loop %iv : i32 in range(%[[tc]]) {
  omp.canonical_loop %iv : i32 in range(%tc) {
    // CHECK-NEXT: llvm.add %iv, %iv : i32
    %newval = llvm.add %iv, %iv: i32
    // CHECK-NEXT: omp.terminator
    omp.terminator
  }
  return
}


// CHECK-LABEL: @omp_canonloop_sequential_pretty(
// CHECK-SAME: %[[tc:.+]]: i32)
func.func @omp_canonloop_sequential_pretty(%tc : i32) -> () {
  // CHECK-NEXT: %canonloop_s0 = omp.new_cli
  %canonloop_s0 = omp.new_cli
  // CHECK-NEXT: omp.canonical_loop(%canonloop_s0) %iv_s0 : i32 in range(%[[tc]]) {
  omp.canonical_loop(%canonloop_s0) %iv_s0 : i32 in range(%tc) {
    // CHECK-NEXT: omp.terminator
    omp.terminator
  }

  // CHECK: %canonloop_s1 = omp.new_cli
  %canonloop_s1 = omp.new_cli
  // CHECK-NEXT: omp.canonical_loop(%canonloop_s1) %iv_s1 : i32 in range(%[[tc]]) {
  omp.canonical_loop(%canonloop_s1) %iv_s1 : i32 in range(%tc) {
    // CHECK-NEXT: omp.terminator
    omp.terminator
  }

  // CHECK: %canonloop_s2 = omp.new_cli
  %canonloop_s2 = omp.new_cli
  // CHECK-NEXT: omp.canonical_loop(%canonloop_s2) %iv_s2 : i32 in range(%[[tc]]) {
  omp.canonical_loop(%canonloop_s2) %iv_s2 : i32 in range(%tc) {
    // CHECK-NEXT: omp.terminator
    omp.terminator
  }

  return
}


// CHECK-LABEL: @omp_canonloop_2d_nested_pretty(
// CHECK-SAME: %[[tc:.+]]: i32)
func.func @omp_canonloop_2d_nested_pretty(%tc : i32) -> () {
  // CHECK-NEXT: %canonloop = omp.new_cli
  %canonloop = omp.new_cli
  // CHECK-NEXT: %canonloop_d1 = omp.new_cli
  %canonloop_d1 = omp.new_cli
  // CHECK-NEXT: omp.canonical_loop(%canonloop) %iv : i32 in range(%[[tc]]) {
  omp.canonical_loop(%canonloop) %iv : i32 in range(%tc) {
    // CHECK-NEXT: omp.canonical_loop(%canonloop_d1) %iv_d1 : i32 in range(%[[tc]]) {
    omp.canonical_loop(%canonloop_d1) %iv_d1 : i32 in range(%tc) {
      // CHECK: omp.terminator
      omp.terminator
    }
    // CHECK: omp.terminator
    omp.terminator
  }
  return
}


// CHECK-LABEL: @omp_canonloop_3d_nested_pretty(
// CHECK-SAME: %[[tc:.+]]: i32)
func.func @omp_canonloop_3d_nested_pretty(%tc : i32) -> () {
  // CHECK: %canonloop = omp.new_cli
  %canonloop = omp.new_cli
  // CHECK: %canonloop_d1 = omp.new_cli
  %canonloop_d1 = omp.new_cli
  // CHECK: %canonloop_d2 = omp.new_cli
  %canonloop_d2 = omp.new_cli
  // CHECK-NEXT: omp.canonical_loop(%canonloop) %iv : i32 in range(%[[tc]]) {
  omp.canonical_loop(%canonloop) %iv : i32 in range(%tc) {
    // CHECK-NEXT: omp.canonical_loop(%canonloop_d1) %iv_d1 : i32 in range(%[[tc]]) {
    omp.canonical_loop(%canonloop_d1) %iv_1d : i32 in range(%tc) {
      // CHECK-NEXT: omp.canonical_loop(%canonloop_d2) %iv_d2 : i32 in range(%[[tc]]) {
      omp.canonical_loop(%canonloop_d2) %iv_d2 : i32 in range(%tc) {
        // CHECK-NEXT: omp.terminator
        omp.terminator
      // CHECK-NEXT: }
      }
      // CHECK-NEXT: omp.terminator
      omp.terminator
    // CHECK-NEXT: }
    }
    // CHECK-NEXT: omp.terminator
    omp.terminator
  }

  return
}


// CHECK-LABEL: @omp_canonloop_sequential_nested_pretty(
// CHECK-SAME: %[[tc:.+]]: i32)
func.func @omp_canonloop_sequential_nested_pretty(%tc : i32) -> () {
  // CHECK-NEXT: %canonloop_s0 = omp.new_cli
  %canonloop_s0 = omp.new_cli
  // CHECK-NEXT: %canonloop_s0_d1 = omp.new_cli
  %canonloop_s0_d1 = omp.new_cli
  // CHECK-NEXT: omp.canonical_loop(%canonloop_s0) %iv_s0 : i32 in range(%[[tc]]) {
  omp.canonical_loop(%canonloop_s0) %iv_s0 : i32 in range(%tc) {
   // CHECK-NEXT: omp.canonical_loop(%canonloop_s0_d1) %iv_s0_d1 : i32 in range(%[[tc]]) {
    omp.canonical_loop(%canonloop_s0_d1) %iv_s0_d1 : i32 in range(%tc) {
      // CHECK-NEXT: omp.terminator
      omp.terminator
    // CHECK-NEXT: }
    }
    // CHECK-NEXT: omp.terminator
    omp.terminator
  // CHECK-NEXT: }
  }

  // CHECK-NEXT: %canonloop_s1 = omp.new_cli
  %canonloop_s1 = omp.new_cli
  // CHECK-NEXT: %canonloop_s1_d1 = omp.new_cli
  %canonloop_s1_d1 = omp.new_cli
  // CHECK-NEXT:  omp.canonical_loop(%canonloop_s1) %iv_s1 : i32 in range(%[[tc]]) {
  omp.canonical_loop(%canonloop_s1) %iv_s1 : i32 in range(%tc) {
    // CHECK-NEXT:  omp.canonical_loop(%canonloop_s1_d1) %iv_s1_d1 : i32 in range(%[[tc]]) {
    omp.canonical_loop(%canonloop_s1_d1) %iv_s1d1 : i32 in range(%tc) {
      // CHECK-NEXT: omp.terminator
      omp.terminator
    // CHECK-NEXT: }
    }
    // CHECK-NEXT: omp.terminator
    omp.terminator
  }

  return
}


// CHECK-LABEL: @omp_newcli_unused(
// CHECK-SAME: )
func.func @omp_newcli_unused() -> () {
  // CHECK-NEXT:  %cli = omp.new_cli
  %cli = omp.new_cli
  // CHECK-NEXT: return
  return
}


// CHECK-LABEL: @omp_canonloop_multiregion_isolatedfromabove(
func.func @omp_canonloop_multiregion_isolatedfromabove() -> () {
  omp.private {type = firstprivate} @x.privatizer : !llvm.ptr init {
    ^bb0(%arg0: !llvm.ptr, %arg1: !llvm.ptr):
      %c42_i32 = arith.constant 42: i32
      // CHECK: omp.canonical_loop %iv : i32 in range(%c42_i32) {
      omp.canonical_loop %iv1 : i32 in range(%c42_i32) {
        omp.terminator
      }
      // CHECK: omp.yield
      omp.yield(%arg0 : !llvm.ptr)
  } copy {
    ^bb0(%arg0: !llvm.ptr, %arg1: !llvm.ptr):
      %c42_i32 = arith.constant 42: i32
      // CHECK: omp.canonical_loop %iv : i32 in range(%c42_i32) {
      omp.canonical_loop %iv : i32 in range(%c42_i32) {
        // CHECK: omp.canonical_loop %iv_d1 : i32 in range(%c42_i32) {
        omp.canonical_loop %iv_d1 : i32 in range(%c42_i32) {
          omp.terminator
        }
        omp.terminator
      }
      // CHECK: omp.yield
      omp.yield(%arg0 : !llvm.ptr)
  } dealloc {
    ^bb0(%arg0: !llvm.ptr):
      %c42_i32 = arith.constant 42: i32
      // CHECK: omp.canonical_loop %iv_s0 : i32 in range(%c42_i32) {
      omp.canonical_loop %iv_s0 : i32 in range(%c42_i32) {
        omp.terminator
      }
      // CHECK: omp.canonical_loop %iv_s1 : i32 in range(%c42_i32) {
      omp.canonical_loop %iv_s1 : i32 in range(%c42_i32) {
        omp.terminator
      }
      // CHECK: omp.yield
      omp.yield
  }

  // CHECK: return
  return
}


// CHECK-LABEL: @omp_canonloop_multiregion(
func.func @omp_canonloop_multiregion(%c : i1) -> () {
  %c42_i32 = arith.constant 42: i32
  %canonloop1 = omp.new_cli
  %canonloop2 = omp.new_cli
  %canonloop3 = omp.new_cli
  scf.if %c {
    // CHECK: omp.canonical_loop(%canonloop_r0) %iv_r0 : i32 in range(%c42_i32) {
    omp.canonical_loop(%canonloop1) %iv1 : i32 in range(%c42_i32) {
      omp.terminator
    }
  } else {
    // CHECK: omp.canonical_loop(%canonloop_r1_s0) %iv_r1_s0 : i32 in range(%c42_i32) {
    omp.canonical_loop(%canonloop2)  %iv2 : i32 in range(%c42_i32) {
      omp.terminator
    }
    // CHECK: omp.canonical_loop(%canonloop_r1_s1) %iv_r1_s1 : i32 in range(%c42_i32) {
    omp.canonical_loop(%canonloop3)  %iv3 : i32 in range(%c42_i32) {
      omp.terminator
    }
  }

  // CHECK: return
  return
}