[mlir][sparse] clean vectorization bail-out for VL=0

aartbik · aartbik · commit 14c0317fef7f · 2023-01-23T12:12:05.000-08:00
Fixes llvm#59970 Reviewed By: Peiming Differential Revision: https://reviews.llvm.org/D142290
diff --git a/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorPasses.cpp b/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorPasses.cpp
@@ -268,6 +268,8 @@ struct SparseVectorizationPass
   }
 
   void runOnOperation() override {
+    if (vectorLength == 0)
+      return signalPassFailure();
     auto *ctx = &getContext();
     RewritePatternSet patterns(ctx);
     populateSparseVectorizationPatterns(

Original file line number	Diff line number	Diff line change
`@@ -268,6 +268,8 @@ struct SparseVectorizationPass`
`268`	`268`	`}`
`269`	`269`
`270`	`270`	`void runOnOperation() override {`
	`271`	`+ if (vectorLength == 0)`
	`272`	`+ return signalPassFailure();`
`271`	`273`	`auto *ctx = &getContext();`
`272`	`274`	`RewritePatternSet patterns(ctx);`
`273`	`275`	`populateSparseVectorizationPatterns(`