[Mlir-commits] [mlir] [mlir][sparse] allow YieldOp to yield	multiple values. (PR #87261)
    Aart Bik 
    llvmlistbot at llvm.org
       
    Mon Apr  1 10:07:11 PDT 2024
    
    
  
================
@@ -1302,14 +1304,24 @@ def SparseTensor_YieldOp : SparseTensor_Op<"yield", [Pure, Terminator]>,
   let builders = [
     OpBuilder<(ins),
     [{
-      build($_builder, $_state, Value());
+      build($_builder, $_state, ValueRange());
+    }]>,
+    OpBuilder<(ins "Value":$yieldVal),
+    [{
+      build($_builder, $_state, ValueRange(yieldVal));
     }]>
   ];
 
+  let extraClassDeclaration = [{
+     Value getSingleResult() {
----------------
aartbik wrote:
bool hasSingleResult() {
  return getResults.size() == 1;
}
Value getSingleResult() {
  assert(hasSingleResult()); 
  return getResults().front();
     }
  
https://github.com/llvm/llvm-project/pull/87261
    
    
More information about the Mlir-commits
mailing list