unroll first iter of checked_ilog loop to save one multiplication

model-checking · Apr 23, 2024 · ad38f9b · ad38f9b
1 parent 7a3093a
commit ad38f9b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/core/src/num/uint_macros.rs b/core/src/num/uint_macros.rs
@@ -1093,9 +1093,12 @@ macro_rules! uint_impl {
         pub const fn checked_ilog(self, base: Self) -> Option<u32> {
             if self <= 0 || base <= 1 {
                 None
+            } else if self < base {
+                Some(0)
             } else {
-                let mut n = 0;
-                let mut r = 1;
+                // Since base >= self, n >= 1
+                let mut n = 1;
+                let mut r = base;
 
                 // Optimization for 128 bit wide integers.
                 if Self::BITS == 128 {