isqrt: assume that isqrt takes half as many bits

rust-lang#89273 (comment)
FedericoStra · Sep 28, 2023 · fcdfd5b · fcdfd5b
1 parent 77f9eae
commit fcdfd5b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/library/core/src/num/uint_macros.rs b/library/core/src/num/uint_macros.rs
@@ -2017,6 +2017,13 @@ macro_rules! uint_impl {
                 one >>= 2;
             }
 
+            // SAFETY: the result is positive and fits in an integer with half as many bits.
+            // Inform the optimizer about it.
+            unsafe {
+                intrinsics::assume(0 < res);
+                intrinsics::assume(res < 1 << (Self::BITS / 2));
+            }
+
             res
         }