Opt following logic

PHILO-HE · Dec 10, 2024 · db818ef · db818ef
1 parent d75d2c1
commit db818ef
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/velox/exec/WindowPartition.cpp b/velox/exec/WindowPartition.cpp
@@ -437,7 +437,23 @@ void WindowPartition::updateKRangeFrameBounds(
       } else {
         // start = lastFoundIndex == -1 ? currentRow
         //                              : std::min(lastFoundIndex, numRows - 1);
-        start = currentRow;
+        if (i == 0) {
+          start = currentRow;
+        } else {
+          auto compareResult = data_->compare(
+              partition_[currentRow - 1],
+              partition_[currentRow],
+              mappedFrameColumn,
+              mappedFrameColumn,
+              flags);
+          if (compareResult <= 0) {
+            start = lastFoundIndex == -1
+                ? currentRow
+                : std::min(lastFoundIndex, numRows - 1);
+          } else {
+            start = currentRow;
+          }
+        }
         end = partition_.size();
       }
       rawFrameBounds[i] = searchFrameValue(