File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
mlir/lib/Dialect/Vector/Transforms Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -642,7 +642,7 @@ class FlattenContiguousRowMajorTransferReadPattern
642
642
// Determinine the first memref dimension to collapse
643
643
int64_t firstDimToCollapse = std::max (
644
644
lastDynIndex (sourceType.getShape ()),
645
- sourceType.getRank () - sourceType.getMaxContiguousTrailingDims ());
645
+ sourceType.getRank () - sourceType.getNumContiguousTrailingDims ());
646
646
647
647
// 1. Collapse the source memref
648
648
Value collapsedSource =
@@ -737,7 +737,7 @@ class FlattenContiguousRowMajorTransferWritePattern
737
737
// Determinine the first memref dimension to collapse
738
738
int64_t firstDimToCollapse = std::max (
739
739
lastDynIndex (sourceType.getShape ()),
740
- sourceType.getRank () - sourceType.getMaxContiguousTrailingDims ());
740
+ sourceType.getRank () - sourceType.getNumContiguousTrailingDims ());
741
741
742
742
// 1. Collapse the source memref
743
743
Value collapsedSource =
You can’t perform that action at this time.
0 commit comments