llvm/test/Transforms/VectorCombine/AArch64/vscale-bitcast-shuffle.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt < %s -passes=vector-combine -S -mtriple=aarch64-- | FileCheck %s --check-prefixes=CHECK
   3
   4 target datalayout = "e-m:e-i8:8:32-i16:16:32-i64:64-i128:128-n32:64-S128"
   5 target triple = "aarch64-unknown-linux-gnu"
   6
   7 ; This test checks we are not crashing with TTI when trying to get shuffle cost.
   8 ; This test also check that shuffle mask <vscale x 4 x i32> zeroinitializer is
   9 ; not narrowed into <0, 1, 0, 1, ...>, which we cannot reason if it's a valid
  10 ; splat or not.
  11
  12 define <vscale x 8 x i16> @bitcast_shuffle(<vscale x 4 x i32> %a) {
  13 ; CHECK-LABEL: @bitcast_shuffle(
  14 ; CHECK-NEXT:    [[I:%.*]] = shufflevector <vscale x 4 x i32> [[A:%.*]], <vscale x 4 x i32> undef, <vscale x 4 x i32> zeroinitializer
  15 ; CHECK-NEXT:    [[R:%.*]] = bitcast <vscale x 4 x i32> [[I]] to <vscale x 8 x i16>
  16 ; CHECK-NEXT:    ret <vscale x 8 x i16> [[R]]
  17 ;
  18   %i = shufflevector <vscale x 4 x i32> %a, <vscale x 4 x i32> undef, <vscale x 4 x i32> zeroinitializer
  19   %r = bitcast <vscale x 4 x i32> %i to <vscale x 8 x i16>
  20   ret <vscale x 8 x i16> %r
  21 }