llvm/test/CodeGen/LoongArch/ctpop-with-lsx.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc --mtriple=loongarch32 -mattr=+lsx < %s | FileCheck %s --check-prefix=LA32
   3 ; RUN: llc --mtriple=loongarch64 -mattr=+lsx < %s | FileCheck %s --check-prefix=LA64
   4
   5 declare i8 @llvm.ctpop.i8(i8)
   6 declare i16 @llvm.ctpop.i16(i16)
   7 declare i32 @llvm.ctpop.i32(i32)
   8 declare i64 @llvm.ctpop.i64(i64)
   9
  10 define i8 @test_ctpop_i8(i8 %a) nounwind {
  11 ; LA32-LABEL: test_ctpop_i8:
  12 ; LA32:       # %bb.0:
  13 ; LA32-NEXT:    andi $a0, $a0, 255
  14 ; LA32-NEXT:    vldi $vr0, 0
  15 ; LA32-NEXT:    vinsgr2vr.w $vr0, $a0, 0
  16 ; LA32-NEXT:    vpcnt.w $vr0, $vr0
  17 ; LA32-NEXT:    vpickve2gr.w $a0, $vr0, 0
  18 ; LA32-NEXT:    ret
  19 ;
  20 ; LA64-LABEL: test_ctpop_i8:
  21 ; LA64:       # %bb.0:
  22 ; LA64-NEXT:    andi $a0, $a0, 255
  23 ; LA64-NEXT:    vldi $vr0, 0
  24 ; LA64-NEXT:    vinsgr2vr.d $vr0, $a0, 0
  25 ; LA64-NEXT:    vpcnt.d $vr0, $vr0
  26 ; LA64-NEXT:    vpickve2gr.d $a0, $vr0, 0
  27 ; LA64-NEXT:    ret
  28   %1 = call i8 @llvm.ctpop.i8(i8 %a)
  29   ret i8 %1
  30 }
  31
  32 define i16 @test_ctpop_i16(i16 %a) nounwind {
  33 ; LA32-LABEL: test_ctpop_i16:
  34 ; LA32:       # %bb.0:
  35 ; LA32-NEXT:    bstrpick.w $a0, $a0, 15, 0
  36 ; LA32-NEXT:    vldi $vr0, 0
  37 ; LA32-NEXT:    vinsgr2vr.w $vr0, $a0, 0
  38 ; LA32-NEXT:    vpcnt.w $vr0, $vr0
  39 ; LA32-NEXT:    vpickve2gr.w $a0, $vr0, 0
  40 ; LA32-NEXT:    ret
  41 ;
  42 ; LA64-LABEL: test_ctpop_i16:
  43 ; LA64:       # %bb.0:
  44 ; LA64-NEXT:    bstrpick.d $a0, $a0, 15, 0
  45 ; LA64-NEXT:    vldi $vr0, 0
  46 ; LA64-NEXT:    vinsgr2vr.d $vr0, $a0, 0
  47 ; LA64-NEXT:    vpcnt.d $vr0, $vr0
  48 ; LA64-NEXT:    vpickve2gr.d $a0, $vr0, 0
  49 ; LA64-NEXT:    ret
  50   %1 = call i16 @llvm.ctpop.i16(i16 %a)
  51   ret i16 %1
  52 }
  53
  54 define i32 @test_ctpop_i32(i32 %a) nounwind {
  55 ; LA32-LABEL: test_ctpop_i32:
  56 ; LA32:       # %bb.0:
  57 ; LA32-NEXT:    vldi $vr0, 0
  58 ; LA32-NEXT:    vinsgr2vr.w $vr0, $a0, 0
  59 ; LA32-NEXT:    vpcnt.w $vr0, $vr0
  60 ; LA32-NEXT:    vpickve2gr.w $a0, $vr0, 0
  61 ; LA32-NEXT:    ret
  62 ;
  63 ; LA64-LABEL: test_ctpop_i32:
  64 ; LA64:       # %bb.0:
  65 ; LA64-NEXT:    bstrpick.d $a0, $a0, 31, 0
  66 ; LA64-NEXT:    vldi $vr0, 0
  67 ; LA64-NEXT:    vinsgr2vr.d $vr0, $a0, 0
  68 ; LA64-NEXT:    vpcnt.d $vr0, $vr0
  69 ; LA64-NEXT:    vpickve2gr.d $a0, $vr0, 0
  70 ; LA64-NEXT:    ret
  71   %1 = call i32 @llvm.ctpop.i32(i32 %a)
  72   ret i32 %1
  73 }
  74
  75 define i64 @test_ctpop_i64(i64 %a) nounwind {
  76 ; LA32-LABEL: test_ctpop_i64:
  77 ; LA32:       # %bb.0:
  78 ; LA32-NEXT:    vldi $vr0, 0
  79 ; LA32-NEXT:    vldi $vr1, 0
  80 ; LA32-NEXT:    vinsgr2vr.w $vr1, $a1, 0
  81 ; LA32-NEXT:    vpcnt.w $vr1, $vr1
  82 ; LA32-NEXT:    vpickve2gr.w $a1, $vr1, 0
  83 ; LA32-NEXT:    vinsgr2vr.w $vr0, $a0, 0
  84 ; LA32-NEXT:    vpcnt.w $vr0, $vr0
  85 ; LA32-NEXT:    vpickve2gr.w $a0, $vr0, 0
  86 ; LA32-NEXT:    add.w $a0, $a0, $a1
  87 ; LA32-NEXT:    move $a1, $zero
  88 ; LA32-NEXT:    ret
  89 ;
  90 ; LA64-LABEL: test_ctpop_i64:
  91 ; LA64:       # %bb.0:
  92 ; LA64-NEXT:    vldi $vr0, 0
  93 ; LA64-NEXT:    vinsgr2vr.d $vr0, $a0, 0
  94 ; LA64-NEXT:    vpcnt.d $vr0, $vr0
  95 ; LA64-NEXT:    vpickve2gr.d $a0, $vr0, 0
  96 ; LA64-NEXT:    ret
  97   %1 = call i64 @llvm.ctpop.i64(i64 %a)
  98   ret i64 %1
  99 }