llvm/test/CodeGen/AArch64/dag-combine-insert-subvector.ll

   1 ; RUN: llc < %s -o /dev/null
   2
   3 ; This regression test is defending against a use of the wrong interface
   4 ; (EVT::getVectorNumElements()) for a scalable vector. This issue
   5 ; appeared in DAGCombiner::visitINSERT_SUBVECTOR because of the use of
   6 ; getVectorNumElements() on scalable types.
   7
   8 target triple = "aarch64-unknown-linux-gnu"
   9 attributes #0 = {"target-features"="+sve"}
  10
  11 declare <16 x float> @llvm.vector.extract.v16f32.nxv4f32(<vscale x 4 x float>, i64)
  12 declare <vscale x 2 x double> @llvm.vector.insert.nxv2f64.v8f64(<vscale x 2 x double>, <8 x double>, i64)
  13
  14 define <vscale x 2 x double> @reproducer_one(<vscale x 4 x float> %vec_a) #0 {
  15   %a = call <16 x float> @llvm.vector.extract.v16f32.nxv4f32(<vscale x 4 x float> %vec_a, i64 0)
  16   %b = bitcast <16 x float> %a to <8 x double>
  17   %retval = call <vscale x 2 x double> @llvm.vector.insert.nxv2f64.v8f64(<vscale x 2 x double> undef, <8 x double> %b, i64 0)
  18   ret <vscale x 2 x double> %retval
  19 }
  20
  21 define <vscale x 2 x double> @reproducer_two(<4 x double> %a, <4 x double> %b) #0 {
  22   %concat = shufflevector <4 x double> %a, <4 x double> %b, <8 x i32> <i32 4, i32 5, i32 6, i32 7, i32 0, i32 1, i32 2, i32 3>
  23   %retval = call <vscale x 2 x double> @llvm.vector.insert.nxv2f64.v8f64(<vscale x 2 x double> undef, <8 x double> %concat, i64 0)
  24   ret <vscale x 2 x double> %retval
  25 }