[Mlir-commits] [mlir] 4643fd2 - [mlir][Linalg] Fix crash when tileSizeComputationFunction is left unspecified

[Nicolas Vasilache]

Author: Nicolas Vasilache
Date: 2021-02-10T22:47:05Z
New Revision: 4643fd27c8a428906b1ed22cb225240a4f52694b

URL: https://github.com/llvm/llvm-project/commit/4643fd27c8a428906b1ed22cb225240a4f52694b
DIFF: https://github.com/llvm/llvm-project/commit/4643fd27c8a428906b1ed22cb225240a4f52694b.diff

LOG: [mlir][Linalg] Fix crash when tileSizeComputationFunction is left unspecified

Added: 
    

Modified: 
    mlir/lib/Dialect/Linalg/Transforms/Tiling.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/Linalg/Transforms/Tiling.cpp b/mlir/lib/Dialect/Linalg/Transforms/Tiling.cpp
index f48100e25dd8..fdad9d879927 100644
--- a/mlir/lib/Dialect/Linalg/Transforms/Tiling.cpp
+++ b/mlir/lib/Dialect/Linalg/Transforms/Tiling.cpp
@@ -472,6 +472,9 @@ Optional<TiledLinalgOp> static tileLinalgOpImpl(
   b.setInsertionPoint(op);
   ScopedContext scope(b, op.getLoc());
 
+  if (!options.tileSizeComputationFunction)
+    return llvm::None;
+  
   // Enforce the convention that "tiling by zero" skips tiling a particular
   // dimension. This convention is significantly simpler to handle instead of
   // adjusting affine maps to account for missing dimensions.