Try using is_val_statically_known in pow

Does not compile.
rust-lang · Jul 12, 2024 · 010c332 · 010c332
1 parent 1faa101
commit 010c332
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 23 deletions.
diff --git a/library/core/src/lib.rs b/library/core/src/lib.rs
@@ -170,6 +170,7 @@
 #![feature(internal_impls_macro)]
 #![feature(ip)]
 #![feature(is_ascii_octdigit)]
+#![feature(is_val_statically_known)]
 #![feature(isqrt)]
 #![feature(link_cfg)]
 #![feature(offset_of_enum)]

diff --git a/library/core/src/num/int_macros.rs b/library/core/src/num/int_macros.rs
@@ -2746,32 +2746,39 @@ macro_rules! int_impl {
         pub const fn pow(self, mut exp: u32) -> Self {
             let mut base = self;
 
-            // Unroll multiplications for small exponent values.
-            // This gives the optimizer a way to efficiently inline call sites
-            // for the most common use cases with constant exponents.
-            // Currently, LLVM is unable to unroll the loop below.
-            match exp {
-                0 => return 1,
-                1 => return base,
-                2 => return base * base,
-                3 => {
-                    let squared = base * base;
-                    return squared * base;
-                }
-                4 => {
-                    let squared = base * base;
-                    return squared * squared;
-                }
-                5 => {
-                    let squared = base * base;
-                    return squared * squared * base;
+            if intrinsics::is_val_statically_known(exp) {
+                // Unroll multiplications for small exponent values.
+                // This gives the optimizer a way to efficiently inline call sites
+                // for the most common use cases with constant exponents.
+                // Currently, LLVM is unable to unroll the loop below.
+                match exp {
+                    0 => return 1,
+                    1 => return base,
+                    2 => return base * base,
+                    3 => {
+                        let squared = base * base;
+                        return squared * base;
+                    }
+                    4 => {
+                        let squared = base * base;
+                        return squared * squared;
+                    }
+                    5 => {
+                        let squared = base * base;
+                        return squared * squared * base;
+                    }
+                    6 => {
+                        let cubed = base * base * base;
+                        return cubed * cubed;
+                    }
+                    _ => {}
                 }
-                6 => {
-                    let cubed = base * base * base;
-                    return cubed * cubed;
+            } else {
+                if exp == 0 {
+                    return 1;
                 }
-                _ => {}
             }
+            debug_assert!(exp != 0);
 
             let mut acc = 1;