[clang] e4872d7 - [SveEmitter] Add builtins for svlen
Sander de Smalen via cfe-commits
cfe-commits at lists.llvm.org
Mon Apr 27 13:28:49 PDT 2020
Author: Sander de Smalen
Date: 2020-04-27T21:27:32+01:00
New Revision: e4872d7f08a1d6f657aaf3bd83b2e3b65fccebd4
URL: https://github.com/llvm/llvm-project/commit/e4872d7f08a1d6f657aaf3bd83b2e3b65fccebd4
DIFF: https://github.com/llvm/llvm-project/commit/e4872d7f08a1d6f657aaf3bd83b2e3b65fccebd4.diff
LOG: [SveEmitter] Add builtins for svlen
The svlen builtins return the number of elements in a vector
and are implemented using `llvm.vscale`.
Reviewed By: efriedma
Differential Revision: https://reviews.llvm.org/D78755
Added:
clang/test/CodeGen/aarch64-sve-intrinsics/acle_sve_len.c
Modified:
clang/include/clang/Basic/arm_sve.td
clang/lib/CodeGen/CGBuiltin.cpp
Removed:
################################################################################
diff --git a/clang/include/clang/Basic/arm_sve.td b/clang/include/clang/Basic/arm_sve.td
index a5cacd2103a8..e77ec24634e9 100644
--- a/clang/include/clang/Basic/arm_sve.td
+++ b/clang/include/clang/Basic/arm_sve.td
@@ -851,6 +851,8 @@ def SVCNTH : SInst<"svcnth", "n", "", MergeNone, "aarch64_sve_cnth", [IsAppendSV
def SVCNTW : SInst<"svcntw", "n", "", MergeNone, "aarch64_sve_cntw", [IsAppendSVALL, IsOverloadNone]>;
def SVCNTD : SInst<"svcntd", "n", "", MergeNone, "aarch64_sve_cntd", [IsAppendSVALL, IsOverloadNone]>;
+def SVLEN : SInst<"svlen[_{d}]", "nd", "csilUcUsUiUlhfd", MergeNone>;
+
////////////////////////////////////////////////////////////////////////////////
// Saturating scalar arithmetic
diff --git a/clang/lib/CodeGen/CGBuiltin.cpp b/clang/lib/CodeGen/CGBuiltin.cpp
index e11791a7a7df..b0e5eebe7b88 100644
--- a/clang/lib/CodeGen/CGBuiltin.cpp
+++ b/clang/lib/CodeGen/CGBuiltin.cpp
@@ -7956,6 +7956,25 @@ Value *CodeGenFunction::EmitAArch64SVEBuiltinExpr(unsigned BuiltinID,
return nullptr;
case SVE::BI__builtin_sve_svpfalse_b:
return ConstantInt::getFalse(Ty);
+
+ case SVE::BI__builtin_sve_svlen_f16:
+ case SVE::BI__builtin_sve_svlen_f32:
+ case SVE::BI__builtin_sve_svlen_f64:
+ case SVE::BI__builtin_sve_svlen_s8:
+ case SVE::BI__builtin_sve_svlen_s16:
+ case SVE::BI__builtin_sve_svlen_s32:
+ case SVE::BI__builtin_sve_svlen_s64:
+ case SVE::BI__builtin_sve_svlen_u8:
+ case SVE::BI__builtin_sve_svlen_u16:
+ case SVE::BI__builtin_sve_svlen_u32:
+ case SVE::BI__builtin_sve_svlen_u64: {
+ SVETypeFlags TF(Builtin->TypeModifier);
+ auto VTy = cast<llvm::VectorType>(getSVEType(TF));
+ auto NumEls = llvm::ConstantInt::get(Ty, VTy->getElementCount().Min);
+
+ Function *F = CGM.getIntrinsic(Intrinsic::vscale, Ty);
+ return Builder.CreateMul(NumEls, Builder.CreateCall(F));
+ }
}
/// Should not happen
diff --git a/clang/test/CodeGen/aarch64-sve-intrinsics/acle_sve_len.c b/clang/test/CodeGen/aarch64-sve-intrinsics/acle_sve_len.c
new file mode 100644
index 000000000000..fc8ebaa18868
--- /dev/null
+++ b/clang/test/CodeGen/aarch64-sve-intrinsics/acle_sve_len.c
@@ -0,0 +1,110 @@
+// RUN: %clang_cc1 -D__ARM_FEATURE_SVE -triple aarch64-none-linux-gnu -target-feature +sve -fallow-half-arguments-and-returns -S -O1 -Werror -Wall -emit-llvm -o - %s | FileCheck %s
+// RUN: %clang_cc1 -D__ARM_FEATURE_SVE -DSVE_OVERLOADED_FORMS -triple aarch64-none-linux-gnu -target-feature +sve -fallow-half-arguments-and-returns -S -O1 -Werror -Wall -emit-llvm -o - %s | FileCheck %s
+
+#include <arm_sve.h>
+
+#ifdef SVE_OVERLOADED_FORMS
+// A simple used,unused... macro, long enough to represent any SVE builtin.
+#define SVE_ACLE_FUNC(A1,A2_UNUSED,A3,A4_UNUSED) A1##A3
+#else
+#define SVE_ACLE_FUNC(A1,A2,A3,A4) A1##A2##A3##A4
+#endif
+
+uint64_t test_svlen_s8(svint8_t op)
+{
+ // CHECK-LABEL: test_svlen_s8
+ // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
+ // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 4
+ // CHECK: ret i64 %[[SHL]]
+ return SVE_ACLE_FUNC(svlen,_s8,,)(op);
+}
+
+uint64_t test_svlen_s16(svint16_t op)
+{
+ // CHECK-LABEL: test_svlen_s16
+ // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
+ // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 3
+ // CHECK: ret i64 %[[SHL]]
+ return SVE_ACLE_FUNC(svlen,_s16,,)(op);
+}
+
+uint64_t test_svlen_s32(svint32_t op)
+{
+ // CHECK-LABEL: test_svlen_s32
+ // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
+ // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 2
+ // CHECK: ret i64 %[[SHL]]
+ return SVE_ACLE_FUNC(svlen,_s32,,)(op);
+}
+
+uint64_t test_svlen_s64(svint64_t op)
+{
+ // CHECK-LABEL: test_svlen_s64
+ // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
+ // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 1
+ // CHECK: ret i64 %[[SHL]]
+ return SVE_ACLE_FUNC(svlen,_s64,,)(op);
+}
+
+uint64_t test_svlen_u8(svuint8_t op)
+{
+ // CHECK-LABEL: test_svlen_u8
+ // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
+ // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 4
+ // CHECK: ret i64 %[[SHL]]
+ return SVE_ACLE_FUNC(svlen,_u8,,)(op);
+}
+
+uint64_t test_svlen_u16(svuint16_t op)
+{
+ // CHECK-LABEL: test_svlen_u16
+ // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
+ // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 3
+ // CHECK: ret i64 %[[SHL]]
+ return SVE_ACLE_FUNC(svlen,_u16,,)(op);
+}
+
+uint64_t test_svlen_u32(svuint32_t op)
+{
+ // CHECK-LABEL: test_svlen_u32
+ // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
+ // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 2
+ // CHECK: ret i64 %[[SHL]]
+ return SVE_ACLE_FUNC(svlen,_u32,,)(op);
+}
+
+uint64_t test_svlen_u64(svuint64_t op)
+{
+ // CHECK-LABEL: test_svlen_u64
+ // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
+ // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 1
+ // CHECK: ret i64 %[[SHL]]
+ return SVE_ACLE_FUNC(svlen,_u64,,)(op);
+}
+
+uint64_t test_svlen_f16(svfloat16_t op)
+{
+ // CHECK-LABEL: test_svlen_f16
+ // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
+ // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 3
+ // CHECK: ret i64 %[[SHL]]
+ return SVE_ACLE_FUNC(svlen,_f16,,)(op);
+}
+
+uint64_t test_svlen_f32(svfloat32_t op)
+{
+ // CHECK-LABEL: test_svlen_f32
+ // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
+ // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 2
+ // CHECK: ret i64 %[[SHL]]
+ return SVE_ACLE_FUNC(svlen,_f32,,)(op);
+}
+
+uint64_t test_svlen_f64(svfloat64_t op)
+{
+ // CHECK-LABEL: test_svlen_f64
+ // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
+ // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 1
+ // CHECK: ret i64 %[[SHL]]
+ return SVE_ACLE_FUNC(svlen,_f64,,)(op);
+}
More information about the cfe-commits
mailing list