[Mlir-commits] [mlir] [mlir][sparse] introduce `sparse_tensor.extract_iteration_space` operation. (PR #88554)
Aart Bik
llvmlistbot at llvm.org
Tue Apr 16 10:11:32 PDT 2024
================
@@ -1430,6 +1430,57 @@ def SparseTensor_ForeachOp : SparseTensor_Op<"foreach",
let hasVerifier = 1;
}
+//===----------------------------------------------------------------------===//
+// Sparse Tensor Iteration Operations.
+//===----------------------------------------------------------------------===//
+
+def ExtractIterSpaceOp : SparseTensor_Op<"extract_iteration_space",
+ [Pure, DeclareOpInterfaceMethods<InferTypeOpInterface>]> {
+
+ let arguments = (ins AnySparseTensor:$tensor,
----------------
aartbik wrote:
Okay scratch this. Team decision was to move all others to arguments/results clauses for readability.
https://github.com/llvm/llvm-project/pull/88554
More information about the Mlir-commits
mailing list