Fix tests

Author: ivanradanov

flang/lib/Optimizer/OpenMP/LowerWorkshare.cpp

@@ -249,7 +249,12 @@ static void parallelizeRegion(Region &sourceRegion, Region &targetRegion,
       if (isSafeToParallelize(&op)) {
         singleBuilder.clone(op, singleMapping);
         if (llvm::all_of(op.getOperands(), [&](Value opr) {
-              return rootMapping.contains(opr);
+              // Either we have already remapped it
+              bool remapped = rootMapping.contains(opr);
+              // Or it is available because it dominates `sr`
+              bool dominates =
+                  di.properlyDominates(opr.getDefiningOp(), &*sr.begin);
+              return remapped || dominates;
             })) {
           // Safe to parallelize operations which have all operands available in
           // the root parallel block can be executed there.

flang/test/Transforms/OpenMP/lower-workshare-correct-parallelize.mlir

@@ -14,3 +14,12 @@ func.func @foo() {
   }
   return
 }
+
+// CHECK:    omp.single nowait
+// CHECK:      fir.allocmem
+// CHECK:      fir.shape
+// CHECK:      fir.declare
+// CHECK:      fir.freemem
+// CHECK:      omp.terminator
+// CHECK:    }
+// CHECK:    omp.barrier

flang/test/Transforms/OpenMP/lower-workshare-todo-cfg-dom.mlir

@@ -1,6 +1,6 @@
 // RUN: fir-opt --lower-workshare --allow-unregistered-dialect %s 2>&1 | FileCheck %s
 
-// CHECK: warning: omp workshare with unstructured control flow currently unsupported and will be serialized.
+// CHECK: warning: omp workshare with unstructured control flow is currently unsupported and will be serialized.
 
 // CHECK: omp.parallel
 // CHECK-NEXT: omp.single

flang/test/Transforms/OpenMP/lower-workshare-todo-cfg.mlir

@@ -1,6 +1,6 @@
 // RUN: fir-opt --lower-workshare --allow-unregistered-dialect %s 2>&1 | FileCheck %s
 
-// CHECK: warning: omp workshare with unstructured control flow currently unsupported and will be serialized.
+// CHECK: warning: omp workshare with unstructured control flow is currently unsupported and will be serialized.
 
 // CHECK: omp.parallel
 // CHECK-NEXT: omp.single