clang/test/CodeGen/aarch64-sve-intrinsics/acle_sve_tbl.c

   1 // NOTE: Assertions have been autogenerated by utils/update_cc_test_checks.py
   2 // REQUIRES: aarch64-registered-target
   3 // RUN: %clang_cc1 -triple aarch64-none-linux-gnu -target-feature +sve -S -disable-O0-optnone -Werror -Wall -emit-llvm -o - %s | opt -S -passes=mem2reg,tailcallelim | FileCheck %s
   4 // RUN: %clang_cc1 -triple aarch64-none-linux-gnu -target-feature +sve -S -disable-O0-optnone -Werror -Wall -emit-llvm -o - -x c++ %s | opt -S -passes=mem2reg,tailcallelim | FileCheck %s -check-prefix=CPP-CHECK
   5 // RUN: %clang_cc1 -DSVE_OVERLOADED_FORMS -triple aarch64-none-linux-gnu -target-feature +sve -S -disable-O0-optnone -Werror -Wall -emit-llvm -o - %s | opt -S -passes=mem2reg,tailcallelim | FileCheck %s
   6 // RUN: %clang_cc1 -DSVE_OVERLOADED_FORMS -triple aarch64-none-linux-gnu -target-feature +sve -S -disable-O0-optnone -Werror -Wall -emit-llvm -o - -x c++ %s | opt -S -passes=mem2reg,tailcallelim | FileCheck %s -check-prefix=CPP-CHECK
   7 // RUN: %clang_cc1 -triple aarch64-none-linux-gnu -target-feature +sve -S -disable-O0-optnone -Werror -Wall -o /dev/null %s
   8 #include <arm_sve.h>
   9
  10 #ifdef SVE_OVERLOADED_FORMS
  11 // A simple used,unused... macro, long enough to represent any SVE builtin.
  12 #define SVE_ACLE_FUNC(A1,A2_UNUSED,A3,A4_UNUSED) A1##A3
  13 #else
  14 #define SVE_ACLE_FUNC(A1,A2,A3,A4) A1##A2##A3##A4
  15 #endif
  16
  17 // CHECK-LABEL: @test_svtbl_s8(
  18 // CHECK-NEXT:  entry:
  19 // CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 16 x i8> @llvm.aarch64.sve.tbl.nxv16i8(<vscale x 16 x i8> [[DATA:%.*]], <vscale x 16 x i8> [[INDICES:%.*]])
  20 // CHECK-NEXT:    ret <vscale x 16 x i8> [[TMP0]]
  21 //
  22 // CPP-CHECK-LABEL: @_Z13test_svtbl_s8u10__SVInt8_tu11__SVUint8_t(
  23 // CPP-CHECK-NEXT:  entry:
  24 // CPP-CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 16 x i8> @llvm.aarch64.sve.tbl.nxv16i8(<vscale x 16 x i8> [[DATA:%.*]], <vscale x 16 x i8> [[INDICES:%.*]])
  25 // CPP-CHECK-NEXT:    ret <vscale x 16 x i8> [[TMP0]]
  26 //
  27 svint8_t test_svtbl_s8(svint8_t data, svuint8_t indices)
  28 {
  29   return SVE_ACLE_FUNC(svtbl,_s8,,)(data, indices);
  30 }
  31
  32 // CHECK-LABEL: @test_svtbl_s16(
  33 // CHECK-NEXT:  entry:
  34 // CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 8 x i16> @llvm.aarch64.sve.tbl.nxv8i16(<vscale x 8 x i16> [[DATA:%.*]], <vscale x 8 x i16> [[INDICES:%.*]])
  35 // CHECK-NEXT:    ret <vscale x 8 x i16> [[TMP0]]
  36 //
  37 // CPP-CHECK-LABEL: @_Z14test_svtbl_s16u11__SVInt16_tu12__SVUint16_t(
  38 // CPP-CHECK-NEXT:  entry:
  39 // CPP-CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 8 x i16> @llvm.aarch64.sve.tbl.nxv8i16(<vscale x 8 x i16> [[DATA:%.*]], <vscale x 8 x i16> [[INDICES:%.*]])
  40 // CPP-CHECK-NEXT:    ret <vscale x 8 x i16> [[TMP0]]
  41 //
  42 svint16_t test_svtbl_s16(svint16_t data, svuint16_t indices)
  43 {
  44   return SVE_ACLE_FUNC(svtbl,_s16,,)(data, indices);
  45 }
  46
  47 // CHECK-LABEL: @test_svtbl_s32(
  48 // CHECK-NEXT:  entry:
  49 // CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 4 x i32> @llvm.aarch64.sve.tbl.nxv4i32(<vscale x 4 x i32> [[DATA:%.*]], <vscale x 4 x i32> [[INDICES:%.*]])
  50 // CHECK-NEXT:    ret <vscale x 4 x i32> [[TMP0]]
  51 //
  52 // CPP-CHECK-LABEL: @_Z14test_svtbl_s32u11__SVInt32_tu12__SVUint32_t(
  53 // CPP-CHECK-NEXT:  entry:
  54 // CPP-CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 4 x i32> @llvm.aarch64.sve.tbl.nxv4i32(<vscale x 4 x i32> [[DATA:%.*]], <vscale x 4 x i32> [[INDICES:%.*]])
  55 // CPP-CHECK-NEXT:    ret <vscale x 4 x i32> [[TMP0]]
  56 //
  57 svint32_t test_svtbl_s32(svint32_t data, svuint32_t indices)
  58 {
  59   return SVE_ACLE_FUNC(svtbl,_s32,,)(data, indices);
  60 }
  61
  62 // CHECK-LABEL: @test_svtbl_s64(
  63 // CHECK-NEXT:  entry:
  64 // CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 2 x i64> @llvm.aarch64.sve.tbl.nxv2i64(<vscale x 2 x i64> [[DATA:%.*]], <vscale x 2 x i64> [[INDICES:%.*]])
  65 // CHECK-NEXT:    ret <vscale x 2 x i64> [[TMP0]]
  66 //
  67 // CPP-CHECK-LABEL: @_Z14test_svtbl_s64u11__SVInt64_tu12__SVUint64_t(
  68 // CPP-CHECK-NEXT:  entry:
  69 // CPP-CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 2 x i64> @llvm.aarch64.sve.tbl.nxv2i64(<vscale x 2 x i64> [[DATA:%.*]], <vscale x 2 x i64> [[INDICES:%.*]])
  70 // CPP-CHECK-NEXT:    ret <vscale x 2 x i64> [[TMP0]]
  71 //
  72 svint64_t test_svtbl_s64(svint64_t data, svuint64_t indices)
  73 {
  74   return SVE_ACLE_FUNC(svtbl,_s64,,)(data, indices);
  75 }
  76
  77 // CHECK-LABEL: @test_svtbl_u8(
  78 // CHECK-NEXT:  entry:
  79 // CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 16 x i8> @llvm.aarch64.sve.tbl.nxv16i8(<vscale x 16 x i8> [[DATA:%.*]], <vscale x 16 x i8> [[INDICES:%.*]])
  80 // CHECK-NEXT:    ret <vscale x 16 x i8> [[TMP0]]
  81 //
  82 // CPP-CHECK-LABEL: @_Z13test_svtbl_u8u11__SVUint8_tS_(
  83 // CPP-CHECK-NEXT:  entry:
  84 // CPP-CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 16 x i8> @llvm.aarch64.sve.tbl.nxv16i8(<vscale x 16 x i8> [[DATA:%.*]], <vscale x 16 x i8> [[INDICES:%.*]])
  85 // CPP-CHECK-NEXT:    ret <vscale x 16 x i8> [[TMP0]]
  86 //
  87 svuint8_t test_svtbl_u8(svuint8_t data, svuint8_t indices)
  88 {
  89   return SVE_ACLE_FUNC(svtbl,_u8,,)(data, indices);
  90 }
  91
  92 // CHECK-LABEL: @test_svtbl_u16(
  93 // CHECK-NEXT:  entry:
  94 // CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 8 x i16> @llvm.aarch64.sve.tbl.nxv8i16(<vscale x 8 x i16> [[DATA:%.*]], <vscale x 8 x i16> [[INDICES:%.*]])
  95 // CHECK-NEXT:    ret <vscale x 8 x i16> [[TMP0]]
  96 //
  97 // CPP-CHECK-LABEL: @_Z14test_svtbl_u16u12__SVUint16_tS_(
  98 // CPP-CHECK-NEXT:  entry:
  99 // CPP-CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 8 x i16> @llvm.aarch64.sve.tbl.nxv8i16(<vscale x 8 x i16> [[DATA:%.*]], <vscale x 8 x i16> [[INDICES:%.*]])
 100 // CPP-CHECK-NEXT:    ret <vscale x 8 x i16> [[TMP0]]
 101 //
 102 svuint16_t test_svtbl_u16(svuint16_t data, svuint16_t indices)
 103 {
 104   return SVE_ACLE_FUNC(svtbl,_u16,,)(data, indices);
 105 }
 106
 107 // CHECK-LABEL: @test_svtbl_u32(
 108 // CHECK-NEXT:  entry:
 109 // CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 4 x i32> @llvm.aarch64.sve.tbl.nxv4i32(<vscale x 4 x i32> [[DATA:%.*]], <vscale x 4 x i32> [[INDICES:%.*]])
 110 // CHECK-NEXT:    ret <vscale x 4 x i32> [[TMP0]]
 111 //
 112 // CPP-CHECK-LABEL: @_Z14test_svtbl_u32u12__SVUint32_tS_(
 113 // CPP-CHECK-NEXT:  entry:
 114 // CPP-CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 4 x i32> @llvm.aarch64.sve.tbl.nxv4i32(<vscale x 4 x i32> [[DATA:%.*]], <vscale x 4 x i32> [[INDICES:%.*]])
 115 // CPP-CHECK-NEXT:    ret <vscale x 4 x i32> [[TMP0]]
 116 //
 117 svuint32_t test_svtbl_u32(svuint32_t data, svuint32_t indices)
 118 {
 119   return SVE_ACLE_FUNC(svtbl,_u32,,)(data, indices);
 120 }
 121
 122 // CHECK-LABEL: @test_svtbl_u64(
 123 // CHECK-NEXT:  entry:
 124 // CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 2 x i64> @llvm.aarch64.sve.tbl.nxv2i64(<vscale x 2 x i64> [[DATA:%.*]], <vscale x 2 x i64> [[INDICES:%.*]])
 125 // CHECK-NEXT:    ret <vscale x 2 x i64> [[TMP0]]
 126 //
 127 // CPP-CHECK-LABEL: @_Z14test_svtbl_u64u12__SVUint64_tS_(
 128 // CPP-CHECK-NEXT:  entry:
 129 // CPP-CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 2 x i64> @llvm.aarch64.sve.tbl.nxv2i64(<vscale x 2 x i64> [[DATA:%.*]], <vscale x 2 x i64> [[INDICES:%.*]])
 130 // CPP-CHECK-NEXT:    ret <vscale x 2 x i64> [[TMP0]]
 131 //
 132 svuint64_t test_svtbl_u64(svuint64_t data, svuint64_t indices)
 133 {
 134   return SVE_ACLE_FUNC(svtbl,_u64,,)(data, indices);
 135 }
 136
 137 // CHECK-LABEL: @test_svtbl_f16(
 138 // CHECK-NEXT:  entry:
 139 // CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 8 x half> @llvm.aarch64.sve.tbl.nxv8f16(<vscale x 8 x half> [[DATA:%.*]], <vscale x 8 x i16> [[INDICES:%.*]])
 140 // CHECK-NEXT:    ret <vscale x 8 x half> [[TMP0]]
 141 //
 142 // CPP-CHECK-LABEL: @_Z14test_svtbl_f16u13__SVFloat16_tu12__SVUint16_t(
 143 // CPP-CHECK-NEXT:  entry:
 144 // CPP-CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 8 x half> @llvm.aarch64.sve.tbl.nxv8f16(<vscale x 8 x half> [[DATA:%.*]], <vscale x 8 x i16> [[INDICES:%.*]])
 145 // CPP-CHECK-NEXT:    ret <vscale x 8 x half> [[TMP0]]
 146 //
 147 svfloat16_t test_svtbl_f16(svfloat16_t data, svuint16_t indices)
 148 {
 149   return SVE_ACLE_FUNC(svtbl,_f16,,)(data, indices);
 150 }
 151
 152 // CHECK-LABEL: @test_svtbl_f32(
 153 // CHECK-NEXT:  entry:
 154 // CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 4 x float> @llvm.aarch64.sve.tbl.nxv4f32(<vscale x 4 x float> [[DATA:%.*]], <vscale x 4 x i32> [[INDICES:%.*]])
 155 // CHECK-NEXT:    ret <vscale x 4 x float> [[TMP0]]
 156 //
 157 // CPP-CHECK-LABEL: @_Z14test_svtbl_f32u13__SVFloat32_tu12__SVUint32_t(
 158 // CPP-CHECK-NEXT:  entry:
 159 // CPP-CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 4 x float> @llvm.aarch64.sve.tbl.nxv4f32(<vscale x 4 x float> [[DATA:%.*]], <vscale x 4 x i32> [[INDICES:%.*]])
 160 // CPP-CHECK-NEXT:    ret <vscale x 4 x float> [[TMP0]]
 161 //
 162 svfloat32_t test_svtbl_f32(svfloat32_t data, svuint32_t indices)
 163 {
 164   return SVE_ACLE_FUNC(svtbl,_f32,,)(data, indices);
 165 }
 166
 167 // CHECK-LABEL: @test_svtbl_f64(
 168 // CHECK-NEXT:  entry:
 169 // CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 2 x double> @llvm.aarch64.sve.tbl.nxv2f64(<vscale x 2 x double> [[DATA:%.*]], <vscale x 2 x i64> [[INDICES:%.*]])
 170 // CHECK-NEXT:    ret <vscale x 2 x double> [[TMP0]]
 171 //
 172 // CPP-CHECK-LABEL: @_Z14test_svtbl_f64u13__SVFloat64_tu12__SVUint64_t(
 173 // CPP-CHECK-NEXT:  entry:
 174 // CPP-CHECK-NEXT:    [[TMP0:%.*]] = tail call <vscale x 2 x double> @llvm.aarch64.sve.tbl.nxv2f64(<vscale x 2 x double> [[DATA:%.*]], <vscale x 2 x i64> [[INDICES:%.*]])
 175 // CPP-CHECK-NEXT:    ret <vscale x 2 x double> [[TMP0]]
 176 //
 177 svfloat64_t test_svtbl_f64(svfloat64_t data, svuint64_t indices)
 178 {
 179   return SVE_ACLE_FUNC(svtbl,_f64,,)(data, indices);
 180 }