Fix bug in DwrfReader onRead

Summary: #9640 We weren't translating to loadUnitIdx. Reviewed By: kgpai, Sullivan-Patrick Differential Revision: D56658575 fbshipit-source-id: 5c293be30a7e43afd67252d291b58d6bd1cb4ae6
facebookincubator · Apr 27, 2024 · 6bcf11e · 6bcf11e
1 parent c2526ba
commit 6bcf11e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/velox/dwio/dwrf/reader/DwrfReader.cpp b/velox/dwio/dwrf/reader/DwrfReader.cpp
@@ -648,7 +648,8 @@ uint64_t DwrfRowReader::next(
   // reading of the data.
   auto strideSize = getReader().getFooter().rowIndexStride();
   strideIndex_ = strideSize > 0 ? currentRowInStripe_ / strideSize : 0;
-  unitLoader_->onRead(currentStripe_, currentRowInStripe_, rowsToRead);
+  const auto loadUnitIdx = currentStripe_ - firstStripe_;
+  unitLoader_->onRead(loadUnitIdx, currentRowInStripe_, rowsToRead);
   readNext(rowsToRead, mutation, result);
   currentRowInStripe_ += rowsToRead;
   return rowsToRead;