diff --git a/llvm/test/CodeGen/AArch64/sve-vector-splat.ll b/llvm/test/CodeGen/AArch64/sve-vector-splat.ll index d2e15f462dad3e..58821738f3a3ad 100644 --- a/llvm/test/CodeGen/AArch64/sve-vector-splat.ll +++ b/llvm/test/CodeGen/AArch64/sve-vector-splat.ll @@ -116,6 +116,17 @@ define @sve_splat_8xi8(i8 %val) { ret %splat } +define @sve_splat_8xi8_imm() { +; CHECK-LABEL: sve_splat_8xi8_imm: +; CHECK: // %bb.0: +; CHECK-NEXT: mov w8, #255 +; CHECK-NEXT: mov z0.h, w8 +; CHECK-NEXT: ret + %ins = insertelement undef, i8 -1, i32 0 + %splat = shufflevector %ins, undef, zeroinitializer + ret %splat +} + define @sve_splat_2xi16(i16 %val) { ; CHECK-LABEL: sve_splat_2xi16: ; CHECK: // %bb.0: @@ -137,6 +148,17 @@ define @sve_splat_4xi16(i16 %val) { ret %splat } +define @sve_splat_4xi16_imm() { +; CHECK-LABEL: sve_splat_4xi16_imm: +; CHECK: // %bb.0: +; CHECK-NEXT: mov w8, #65535 +; CHECK-NEXT: mov z0.s, w8 +; CHECK-NEXT: ret + %ins = insertelement undef, i16 -1, i32 0 + %splat = shufflevector %ins, undef, zeroinitializer + ret %splat +} + define @sve_splat_2xi32(i32 %val) { ; CHECK-LABEL: sve_splat_2xi32: ; CHECK: // %bb.0: @@ -148,6 +170,17 @@ define @sve_splat_2xi32(i32 %val) { ret %splat } +define @sve_splat_2xi32_imm() { +; CHECK-LABEL: sve_splat_2xi32_imm: +; CHECK: // %bb.0: +; CHECK-NEXT: mov w8, #-1 +; CHECK-NEXT: mov z0.d, x8 +; CHECK-NEXT: ret + %ins = insertelement undef, i32 -1, i32 0 + %splat = shufflevector %ins, undef, zeroinitializer + ret %splat +} + ;; Widen/split splats of wide vector types. define @sve_splat_1xi32(i32 %val) { @@ -497,9 +530,9 @@ define @splat_nxv4f32_imm_out_of_range() { define @splat_nxv2f64_imm_out_of_range() { ; CHECK-LABEL: splat_nxv2f64_imm_out_of_range: ; CHECK: // %bb.0: -; CHECK-NEXT: adrp x8, .LCPI47_0 +; CHECK-NEXT: adrp x8, .LCPI50_0 ; CHECK-NEXT: ptrue p0.d -; CHECK-NEXT: add x8, x8, :lo12:.LCPI47_0 +; CHECK-NEXT: add x8, x8, :lo12:.LCPI50_0 ; CHECK-NEXT: ld1rd { z0.d }, p0/z, [x8] ; CHECK-NEXT: ret %1 = insertelement undef, double 3.33, i32 0