[Mlir-commits] [mlir] [mlir][sparse] implement lowering rules for IterateOp. (PR #95286)

Aart Bik llvmlistbot at llvm.org
Fri Jun 14 10:33:17 PDT 2024


================
@@ -34,6 +34,19 @@ convertIterSpaceType(IterSpaceType itSp, SmallVectorImpl<Type> &fields) {
   return success();
 }
 
+static std::optional<LogicalResult>
+convertIteratorType(IteratorType itTp, SmallVectorImpl<Type> &fields) {
+  // The actually Iterator Values (that are updated every iteration).
+  auto idxTp = IndexType::get(itTp.getContext());
+  // TODO: This assumes there is no batch dimenstion in the sparse tensor.
----------------
aartbik wrote:

also, is there a way to assert on this?

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


More information about the Mlir-commits mailing list