llvm/test/Transforms/SLPVectorizer/RISCV/trunc-bv-multi-uses.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 4
   2 ; RUN: opt -S --passes=slp-vectorizer -mtriple=riscv64-unknown-linux-gnu -mattr="+v" < %s -slp-threshold=-10 | FileCheck %s
   3
   4 define i32 @test(i64 %v1, i64 %v2) {
   5 ; CHECK-LABEL: define i32 @test(
   6 ; CHECK-SAME: i64 [[V1:%.*]], i64 [[V2:%.*]]) #[[ATTR0:[0-9]+]] {
   7 ; CHECK-NEXT:  entry:
   8 ; CHECK-NEXT:    [[TMP0:%.*]] = insertelement <2 x i64> poison, i64 [[V1]], i32 0
   9 ; CHECK-NEXT:    [[TMP1:%.*]] = insertelement <2 x i64> [[TMP0]], i64 [[V2]], i32 1
  10 ; CHECK-NEXT:    [[TMP2:%.*]] = trunc <2 x i64> [[TMP1]] to <2 x i32>
  11 ; CHECK-NEXT:    [[TMP4:%.*]] = lshr <2 x i64> [[TMP1]], <i64 32, i64 32>
  12 ; CHECK-NEXT:    [[TMP5:%.*]] = trunc <2 x i64> [[TMP4]] to <2 x i32>
  13 ; CHECK-NEXT:    [[TMP6:%.*]] = add <2 x i32> [[TMP2]], [[TMP5]]
  14 ; CHECK-NEXT:    [[TMP7:%.*]] = extractelement <2 x i32> [[TMP6]], i32 0
  15 ; CHECK-NEXT:    [[TMP8:%.*]] = extractelement <2 x i32> [[TMP6]], i32 1
  16 ; CHECK-NEXT:    [[MUL:%.*]] = mul i32 [[TMP7]], [[TMP8]]
  17 ; CHECK-NEXT:    ret i32 [[MUL]]
  18 ;
  19 entry:
  20   %t1 = trunc i64 %v1 to i32
  21   %t2 = trunc i64 %v2 to i32
  22   %lshr1 = lshr i64 %v1, 32
  23   %lshr2 = lshr i64 %v2, 32
  24   %t3 = trunc i64 %lshr1 to i32
  25   %t4 = trunc i64 %lshr2 to i32
  26   %add1 = add i32 %t1, %t3
  27   %add2 = add i32 %t2, %t4
  28   %mul = mul i32 %add1, %add2
  29   ret i32 %mul
  30 }