llvm/test/CodeGen/LoongArch/lsx/intrinsic-sra.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc --mtriple=loongarch64 --mattr=+lsx < %s | FileCheck %s
   3
   4 declare <16 x i8> @llvm.loongarch.lsx.vsra.b(<16 x i8>, <16 x i8>)
   5
   6 define <16 x i8> @lsx_vsra_b(<16 x i8> %va, <16 x i8> %vb) nounwind {
   7 ; CHECK-LABEL: lsx_vsra_b:
   8 ; CHECK:       # %bb.0: # %entry
   9 ; CHECK-NEXT:    vsra.b $vr0, $vr0, $vr1
  10 ; CHECK-NEXT:    ret
  11 entry:
  12   %res = call <16 x i8> @llvm.loongarch.lsx.vsra.b(<16 x i8> %va, <16 x i8> %vb)
  13   ret <16 x i8> %res
  14 }
  15
  16 declare <8 x i16> @llvm.loongarch.lsx.vsra.h(<8 x i16>, <8 x i16>)
  17
  18 define <8 x i16> @lsx_vsra_h(<8 x i16> %va, <8 x i16> %vb) nounwind {
  19 ; CHECK-LABEL: lsx_vsra_h:
  20 ; CHECK:       # %bb.0: # %entry
  21 ; CHECK-NEXT:    vsra.h $vr0, $vr0, $vr1
  22 ; CHECK-NEXT:    ret
  23 entry:
  24   %res = call <8 x i16> @llvm.loongarch.lsx.vsra.h(<8 x i16> %va, <8 x i16> %vb)
  25   ret <8 x i16> %res
  26 }
  27
  28 declare <4 x i32> @llvm.loongarch.lsx.vsra.w(<4 x i32>, <4 x i32>)
  29
  30 define <4 x i32> @lsx_vsra_w(<4 x i32> %va, <4 x i32> %vb) nounwind {
  31 ; CHECK-LABEL: lsx_vsra_w:
  32 ; CHECK:       # %bb.0: # %entry
  33 ; CHECK-NEXT:    vsra.w $vr0, $vr0, $vr1
  34 ; CHECK-NEXT:    ret
  35 entry:
  36   %res = call <4 x i32> @llvm.loongarch.lsx.vsra.w(<4 x i32> %va, <4 x i32> %vb)
  37   ret <4 x i32> %res
  38 }
  39
  40 declare <2 x i64> @llvm.loongarch.lsx.vsra.d(<2 x i64>, <2 x i64>)
  41
  42 define <2 x i64> @lsx_vsra_d(<2 x i64> %va, <2 x i64> %vb) nounwind {
  43 ; CHECK-LABEL: lsx_vsra_d:
  44 ; CHECK:       # %bb.0: # %entry
  45 ; CHECK-NEXT:    vsra.d $vr0, $vr0, $vr1
  46 ; CHECK-NEXT:    ret
  47 entry:
  48   %res = call <2 x i64> @llvm.loongarch.lsx.vsra.d(<2 x i64> %va, <2 x i64> %vb)
  49   ret <2 x i64> %res
  50 }
  51
  52 declare <16 x i8> @llvm.loongarch.lsx.vsrai.b(<16 x i8>, i32)
  53
  54 define <16 x i8> @lsx_vsrai_b(<16 x i8> %va) nounwind {
  55 ; CHECK-LABEL: lsx_vsrai_b:
  56 ; CHECK:       # %bb.0: # %entry
  57 ; CHECK-NEXT:    vsrai.b $vr0, $vr0, 7
  58 ; CHECK-NEXT:    ret
  59 entry:
  60   %res = call <16 x i8> @llvm.loongarch.lsx.vsrai.b(<16 x i8> %va, i32 7)
  61   ret <16 x i8> %res
  62 }
  63
  64 declare <8 x i16> @llvm.loongarch.lsx.vsrai.h(<8 x i16>, i32)
  65
  66 define <8 x i16> @lsx_vsrai_h(<8 x i16> %va) nounwind {
  67 ; CHECK-LABEL: lsx_vsrai_h:
  68 ; CHECK:       # %bb.0: # %entry
  69 ; CHECK-NEXT:    vsrai.h $vr0, $vr0, 15
  70 ; CHECK-NEXT:    ret
  71 entry:
  72   %res = call <8 x i16> @llvm.loongarch.lsx.vsrai.h(<8 x i16> %va, i32 15)
  73   ret <8 x i16> %res
  74 }
  75
  76 declare <4 x i32> @llvm.loongarch.lsx.vsrai.w(<4 x i32>, i32)
  77
  78 define <4 x i32> @lsx_vsrai_w(<4 x i32> %va) nounwind {
  79 ; CHECK-LABEL: lsx_vsrai_w:
  80 ; CHECK:       # %bb.0: # %entry
  81 ; CHECK-NEXT:    vsrai.w $vr0, $vr0, 31
  82 ; CHECK-NEXT:    ret
  83 entry:
  84   %res = call <4 x i32> @llvm.loongarch.lsx.vsrai.w(<4 x i32> %va, i32 31)
  85   ret <4 x i32> %res
  86 }
  87
  88 declare <2 x i64> @llvm.loongarch.lsx.vsrai.d(<2 x i64>, i32)
  89
  90 define <2 x i64> @lsx_vsrai_d(<2 x i64> %va) nounwind {
  91 ; CHECK-LABEL: lsx_vsrai_d:
  92 ; CHECK:       # %bb.0: # %entry
  93 ; CHECK-NEXT:    vsrai.d $vr0, $vr0, 63
  94 ; CHECK-NEXT:    ret
  95 entry:
  96   %res = call <2 x i64> @llvm.loongarch.lsx.vsrai.d(<2 x i64> %va, i32 63)
  97   ret <2 x i64> %res
  98 }