[Mlir-commits] [mlir] [mlir][spirv] Allow yielding values from selection regions (PR #133702)

[Jakub Kuderski]

================
@@ -447,6 +447,14 @@ LogicalResult Serializer::processSelectionOp(spirv::SelectionOp selectionOp) {
   auto mergeID = getBlockID(mergeBlock);
   auto loc = selectionOp.getLoc();
 
+  // Before we do anything wire yielded values with the result of the selection
+  // operation. The selection op is being flatten so we do not have to worry
+  // about values being defined inside a region and used outside it anymore.
+  auto mergeOp = cast<spirv::MergeOp>(mergeBlock->back());
+  assert(selectionOp.getNumResults() == mergeOp.getNumOperands());
+  for (unsigned i = 0; i < selectionOp.getNumResults(); ++i)
----------------
kuhar wrote:

also here

https://github.com/llvm/llvm-project/pull/133702