[Mlir-commits] [mlir] 14c0317 - [mlir][sparse] clean vectorization bail-out for VL=0

[Aart Bik]

Author: Aart Bik
Date: 2023-01-23T12:12:05-08:00
New Revision: 14c0317fef7fa2c3f153f18c33e399660227a16f

URL: https://github.com/llvm/llvm-project/commit/14c0317fef7fa2c3f153f18c33e399660227a16f
DIFF: https://github.com/llvm/llvm-project/commit/14c0317fef7fa2c3f153f18c33e399660227a16f.diff

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

Fixes https://github.com/llvm/llvm-project/issues/59970

Reviewed By: Peiming

Differential Revision: https://reviews.llvm.org/D142290

Added: 
    

Modified: 
    mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorPasses.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorPasses.cpp b/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorPasses.cpp
index c384d6ad966c9..fc4af709d5d8c 100644
--- 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(