continue optimize.

apache · Aug 30, 2024 · c9f0b06 · c9f0b06
1 parent 5bb1805
commit c9f0b06
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/datafusion/functions-aggregate-common/src/aggregate/groups_accumulator.rs b/datafusion/functions-aggregate-common/src/aggregate/groups_accumulator.rs
@@ -473,9 +473,9 @@ impl BlockedGroupIndex {
         let data = raw_index & GROUP_INDEX_DATA_MASK;
         let (highs, lows) = ((data >> 32) as u32, data as u32);
 
-        let block_id = highs * flag as u32;
+        let block_id = (highs & (u32::MAX.wrapping_add(1 - flag as u32))) as u32;
         let block_offset = {
-            let offset_high = highs as u64 * (1 - flag);
+            let offset_high = highs as u64 & (u64::MAX.wrapping_add(flag));
             (offset_high << 32) | (lows as u64)
         };