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
642642 // Determinine the first memref dimension to collapse
643643 int64_t firstDimToCollapse = std::max (
644644 lastDynIndex (sourceType.getShape ()),
645- sourceType.getRank () - sourceType.getMaxContiguousTrailingDims ());
645+ sourceType.getRank () - sourceType.getNumContiguousTrailingDims ());
646646
647647 // 1. Collapse the source memref
648648 Value collapsedSource =
@@ -737,7 +737,7 @@ class FlattenContiguousRowMajorTransferWritePattern
737737 // Determinine the first memref dimension to collapse
738738 int64_t firstDimToCollapse = std::max (
739739 lastDynIndex (sourceType.getShape ()),
740- sourceType.getRank () - sourceType.getMaxContiguousTrailingDims ());
740+ sourceType.getRank () - sourceType.getNumContiguousTrailingDims ());
741741
742742 // 1. Collapse the source memref
743743 Value collapsedSource =
You can’t perform that action at this time.
0 commit comments