Initial commit

PHILO-HE · Feb 27, 2024 · 1c8ebac · 1c8ebac
1 parent 86b4add
commit 1c8ebac
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/velox/exec/Window.cpp b/velox/exec/Window.cpp
@@ -300,8 +300,16 @@ void Window::updateKRowsFrameBounds(
   if (frameArg.index == kConstantChannel) {
     auto constantOffset = frameArg.constant.value();
     auto startValue =
-        startRow + (isKPreceding ? -constantOffset : constantOffset);
-    std::iota(rawFrameBounds, rawFrameBounds + numRows, startValue);
+        (isKPreceding ? -constantOffset : constantOffset) + startRow;
+    // std::iota(rawFrameBounds, rawFrameBounds + numRows, startValue);
+    for (int i = 0; i < numRows; i++) {
+      if (startValue != (int32_t)startValue) {
+        std::fill(rawFrameBounds + i, rawFrameBounds + numRows, startValue - 1);
+      } else {
+        *(rawFrameBounds + i) = startValue;
+      }
+      startValue = startValue + 1;
+    }
   } else {
     currentPartition_->extractColumn(
         frameArg.index, partitionOffset_, numRows, 0, frameArg.value);