[Mlir-commits] [mlir] e828655 - [mlir][linalg] Fix interchange initialization in fusion on tensors.

[Tobias Gysi]

Author: Tobias Gysi
Date: 2021-09-22T17:45:54Z
New Revision: e828655313db901c733bf4bab6ac78c18833d765

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

LOG: [mlir][linalg] Fix interchange initialization in fusion on tensors.

If no interchange vector is given initialize it with the identity permutation from 0 to number of loops.

Reviewed By: mravishankar

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

Added: 
    

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

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/Linalg/Transforms/FusionOnTensors.cpp b/mlir/lib/Dialect/Linalg/Transforms/FusionOnTensors.cpp
index 973d6136738f..31e53f7cf93d 100644
--- a/mlir/lib/Dialect/Linalg/Transforms/FusionOnTensors.cpp
+++ b/mlir/lib/Dialect/Linalg/Transforms/FusionOnTensors.cpp
@@ -420,7 +420,7 @@ struct LinalgTileAndFuseTensorOps
         tileSizes.begin(), tileSizes.begin() + rootOp.getNumLoops());
     SmallVector<int64_t> rootInterchange =
         tileInterchange.empty()
-            ? llvm::to_vector<6>(llvm::seq<int64_t>(0, tileSizes.size()))
+            ? llvm::to_vector<6>(llvm::seq<int64_t>(0, rootOp.getNumLoops()))
             : SmallVector<int64_t>(tileInterchange.begin(),
                                    tileInterchange.begin() +
                                        rootOp.getNumLoops());