Fix imm passing

Author: Licenser

crates/core_arch/src/arm/neon/mod.rs

@@ -1195,8 +1195,8 @@ pub unsafe fn vget_lane_u64(v: uint64x1_t, imm5: i32) -> u64 {
 #[target_feature(enable = "neon")]
 #[cfg_attr(target_arch = "arm", target_feature(enable = "v7"))]
 #[rustc_args_required_const(1)]
-#[cfg_attr(all(test, target_arch = "arm"), assert_instr("vmov.u16"))]
-#[cfg_attr(all(test, target_arch = "aarch64"), assert_instr(umov))]
+#[cfg_attr(all(test, target_arch = "arm"), assert_instr("vmov.u16", imm5 = 2))]
+#[cfg_attr(all(test, target_arch = "aarch64"), assert_instr(umov, imm5 = 2))]
 pub unsafe fn vgetq_lane_u16(v: uint16x8_t, imm5: i32) -> u16 {
     if (imm5) < 0 || (imm5) > 7 {
         unreachable_unchecked()
@@ -1210,8 +1210,8 @@ pub unsafe fn vgetq_lane_u16(v: uint16x8_t, imm5: i32) -> u16 {
 #[target_feature(enable = "neon")]
 #[cfg_attr(target_arch = "arm", target_feature(enable = "v7"))]
 #[rustc_args_required_const(1)]
-#[cfg_attr(all(test, target_arch = "arm"), assert_instr("vmov.u32"))]
-#[cfg_attr(all(test, target_arch = "aarch64"), assert_instr(umov))]
+#[cfg_attr(all(test, target_arch = "arm"), assert_instr("vmov.u32", imm5 = 2))]
+#[cfg_attr(all(test, target_arch = "aarch64"), assert_instr(umov, imm5 = 2))]
 pub unsafe fn vgetq_lane_u32(v: uint32x4_t, imm5: i32) -> u32 {
     if (imm5) < 0 || (imm5) > 3 {
         unreachable_unchecked()
@@ -1225,8 +1225,8 @@ pub unsafe fn vgetq_lane_u32(v: uint32x4_t, imm5: i32) -> u32 {
 #[target_feature(enable = "neon")]
 #[cfg_attr(target_arch = "arm", target_feature(enable = "v7"))]
 #[rustc_args_required_const(1)]
-#[cfg_attr(all(test, target_arch = "arm"), assert_instr("mov.u8"))]
-#[cfg_attr(all(test, target_arch = "aarch64"), assert_instr(umov))]
+#[cfg_attr(all(test, target_arch = "arm"), assert_instr("mov.u8", imm5 = 2))]
+#[cfg_attr(all(test, target_arch = "aarch64"), assert_instr(umov, imm5 = 2))]
 pub unsafe fn vget_lane_u8(v: uint8x8_t, imm5: i32) -> u8 {
     if (imm5) < 0 || (imm5) > 7 {
         unreachable_unchecked()
@@ -1330,8 +1330,7 @@ pub unsafe fn vreinterpretq_u8_s8(a: int8x16_t) -> uint8x16_t {
 #[target_feature(enable = "neon")]
 #[cfg_attr(target_arch = "arm", target_feature(enable = "v7"))]
 #[cfg_attr(all(test, target_arch = "arm"), assert_instr("vshr.s8", imm3 = 1))]
-#[cfg_attr(all(test, target_arch = "aarch64"), assert_instr(shr, n = 3))]
-
+#[cfg_attr(all(test, target_arch = "aarch64"), assert_instr(shr, imm3 = 1))]
 #[rustc_args_required_const(1)]
 pub unsafe fn vshrq_n_u8(a: uint8x16_t, imm3: i32) -> uint8x16_t {
     if imm3 < 0 || imm3 > 7 {
@@ -1363,7 +1362,7 @@ pub unsafe fn vshrq_n_u8(a: uint8x16_t, imm3: i32) -> uint8x16_t {
 #[target_feature(enable = "neon")]
 #[cfg_attr(target_arch = "arm", target_feature(enable = "v7"))]
 #[cfg_attr(all(test, target_arch = "arm"), assert_instr("vshl.s8", imm3 = 1))]
-#[cfg_attr(all(test, target_arch = "aarch64"), assert_instr(shl, n = 3))]
+#[cfg_attr(all(test, target_arch = "aarch64"), assert_instr(shl, imm3 = 1))]
 #[rustc_args_required_const(1)]
 pub unsafe fn vshlq_n_u8(a: uint8x16_t, imm3: i32) -> uint8x16_t {
     if imm3 < 0 || imm3 > 7 {