// RUN: fir-opt --split-input-file --lower-workshare --allow-unregistered-dialect %s | FileCheck %s

// Checks that the omp.workshare.loop_wrapper binds to the correct omp.workshare

func.func @wsfunc() {
  %c1 = arith.constant 1 : index
  %c42 = arith.constant 42 : index
  omp.parallel {
    omp.workshare nowait {
      omp.parallel {
        omp.workshare nowait {
          omp.workshare.loop_wrapper {
            omp.loop_nest (%arg1) : index = (%c1) to (%c42) inclusive step (%c1) {
              "test.test2"() : () -> ()
              omp.yield
            }
          }
          omp.terminator
        }
        omp.terminator
      }
      omp.terminator
    }
    omp.terminator
  }
  return
}

// CHECK-LABEL:   func.func @wsfunc() {
// CHECK:           %[[VAL_0:.*]] = arith.constant 1 : index
// CHECK:           %[[VAL_1:.*]] = arith.constant 42 : index
// CHECK:           omp.parallel {
// CHECK:             omp.single nowait {
// CHECK:               omp.parallel {
// CHECK:                 omp.wsloop nowait {
// CHECK:                   omp.loop_nest (%[[VAL_2:.*]]) : index = (%[[VAL_0]]) to (%[[VAL_1]]) inclusive step (%[[VAL_0]]) {
// CHECK:                     "test.test2"() : () -> ()
// CHECK:                     omp.yield
// CHECK:                   }
// CHECK:                 }
// CHECK:                 omp.terminator
// CHECK:               }
// CHECK:               omp.terminator
// CHECK:             }
// CHECK:             omp.terminator
// CHECK:           }
// CHECK:           return
// CHECK:         }