llvm/test/CodeGen/AArch64/arm64-uzp2-combine.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc < %s -mtriple aarch64-none-linux-gnu | FileCheck %s
   3
   4 ; Test the (concat_vectors (trunc (lshr)), (trunc (lshr))) pattern.
   5
   6 define <16 x i8> @test_combine_v8i16_to_v16i8(<8 x i16> %x, <8 x i16> %y) {
   7 ; CHECK-LABEL: test_combine_v8i16_to_v16i8:
   8 ; CHECK:       // %bb.0: // %entry
   9 ; CHECK-NEXT:    uzp2 v0.16b, v0.16b, v1.16b
  10 ; CHECK-NEXT:    ret
  11 entry:
  12   %lshr1 = lshr <8 x i16> %x, <i16 8, i16 8, i16 8, i16 8, i16 8, i16 8, i16 8, i16 8>
  13   %trunc1 = trunc <8 x i16> %lshr1 to <8 x i8>
  14   %lshr2 = lshr <8 x i16> %y, <i16 8, i16 8, i16 8, i16 8, i16 8, i16 8, i16 8, i16 8>
  15   %trunc2 = trunc <8 x i16> %lshr2 to <8 x i8>
  16   %shuffle = shufflevector <8 x i8> %trunc1, <8 x i8> %trunc2, <16 x i32> <i32 0, i32 1, i32 2, i32 3, i32 4, i32 5, i32 6, i32 7, i32 8, i32 9, i32 10, i32 11, i32 12, i32 13, i32 14, i32 15>
  17   ret <16 x i8> %shuffle
  18 }
  19
  20 define <8 x i16> @test_combine_v4i32_to_v8i16(<4 x i32> %x, <4 x i32> %y) {
  21 ; CHECK-LABEL: test_combine_v4i32_to_v8i16:
  22 ; CHECK:       // %bb.0: // %entry
  23 ; CHECK-NEXT:    uzp2 v0.8h, v0.8h, v1.8h
  24 ; CHECK-NEXT:    ret
  25 entry:
  26   %lshr1 = lshr <4 x i32> %x, <i32 16, i32 16, i32 16, i32 16>
  27   %trunc1 = trunc <4 x i32> %lshr1 to <4 x i16>
  28   %lshr2 = lshr <4 x i32> %y, <i32 16, i32 16, i32 16, i32 16>
  29   %trunc2 = trunc <4 x i32> %lshr2 to <4 x i16>
  30   %shuffle = shufflevector <4 x i16> %trunc1, <4 x i16> %trunc2, <8 x i32> <i32 0, i32 1, i32 2, i32 3, i32 4, i32 5, i32 6, i32 7>
  31   ret <8 x i16> %shuffle
  32 }
  33
  34 define <4 x i32> @test_combine_v2i64_to_v4i32(<2 x i64> %x, <2 x i64> %y) {
  35 ; CHECK-LABEL: test_combine_v2i64_to_v4i32:
  36 ; CHECK:       // %bb.0: // %entry
  37 ; CHECK-NEXT:    uzp2 v0.4s, v0.4s, v1.4s
  38 ; CHECK-NEXT:    ret
  39 entry:
  40   %lshr1 = lshr <2 x i64> %x, <i64 32, i64 32>
  41   %trunc1 = trunc <2 x i64> %lshr1 to <2 x i32>
  42   %lshr2 = lshr <2 x i64> %y, <i64 32, i64 32>
  43   %trunc2 = trunc <2 x i64> %lshr2 to <2 x i32>
  44   %shuffle = shufflevector <2 x i32> %trunc1, <2 x i32> %trunc2, <4 x i32> <i32 0, i32 1, i32 2, i32 3>
  45   ret <4 x i32> %shuffle
  46 }
  47