small fix for const generic used for static_assert_imm2

Author: luojia65

crates/core_arch/src/riscv_shared/mod.rs

@@ -684,7 +684,7 @@ pub fn sm3p1(x: u32) -> u32 {
 ///
 /// According to RISC-V Cryptography Extensions, Volume I, the execution latency of
 /// this instruction must always be independent from the data it operates on.
-pub fn sm4ed<const BS: u8>(x: u32, a: u32) -> u32 {
+pub fn sm4ed<const BS: i32>(x: u32, a: u32) -> u32 {
     static_assert_imm2!(BS); // `bs` immediate value must be within [0, 3]
     let ans: u32;
     match BS {
@@ -749,7 +749,7 @@ pub fn sm4ed<const BS: u8>(x: u32, a: u32) -> u32 {
 ///
 /// According to RISC-V Cryptography Extensions, Volume I, the execution latency of
 /// this instruction must always be independent from the data it operates on.
-pub fn sm4ks<const BS: u8>(x: u32, k: u32) -> u32 {
+pub fn sm4ks<const BS: i32>(x: u32, k: u32) -> u32 {
     static_assert_imm2!(BS); // `bs` immediate value must be within [0, 3]
     let ans: u32;
     match BS {