Skip to content

Commit 1e344e0

Browse files
authored
[CIR][CIRGen][Builtin][Neon] Lower neon_vaddlvq_s16 (#1231)
Combined implementaiton with `neon_vaddlvq_u16` OG somehow implemented them separately but they are no different except signess and intrinsic name [OG's neon_vaddlvq_s16](https://github.com/llvm/clangir/blob/2b1a638ea07ca10c5727ea835bfbe17b881175cc/clang/lib/CodeGen/CGBuiltin.cpp#L13483) [OG's neon_vaddlvq_u16](https://github.com/llvm/clangir/blob/2b1a638ea07ca10c5727ea835bfbe17b881175cc/clang/lib/CodeGen/CGBuiltin.cpp#L13449)
1 parent 9c0639b commit 1e344e0

File tree

2 files changed

+20
-7
lines changed

2 files changed

+20
-7
lines changed

clang/lib/CIR/CodeGen/CIRGenBuiltinAArch64.cpp

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4325,11 +4325,16 @@ CIRGenFunction::emitAArch64BuiltinExpr(unsigned BuiltinID, const CallExpr *E,
43254325
case NEON::BI__builtin_neon_vaddlvq_u8: {
43264326
llvm_unreachable("NEON::BI__builtin_neon_vaddlvq_u8 NYI");
43274327
}
4328-
case NEON::BI__builtin_neon_vaddlvq_u16: {
4329-
mlir::Type argTy = cir::VectorType::get(builder.getContext(), UInt16Ty, 8);
4328+
case NEON::BI__builtin_neon_vaddlvq_u16:
4329+
usgn = true;
4330+
[[fallthrough]];
4331+
case NEON::BI__builtin_neon_vaddlvq_s16: {
4332+
mlir::Type argTy = cir::VectorType::get(builder.getContext(),
4333+
usgn ? UInt16Ty : SInt16Ty, 8);
43304334
llvm::SmallVector<mlir::Value, 1> argOps = {emitScalarExpr(E->getArg(0))};
4331-
return emitNeonCall(builder, {argTy}, argOps, "aarch64.neon.uaddlv",
4332-
UInt32Ty, getLoc(E->getExprLoc()));
4335+
return emitNeonCall(builder, {argTy}, argOps,
4336+
usgn ? "aarch64.neon.uaddlv" : "aarch64.neon.saddlv",
4337+
usgn ? UInt32Ty : SInt32Ty, getLoc(E->getExprLoc()));
43334338
}
43344339
case NEON::BI__builtin_neon_vaddlv_s8: {
43354340
llvm_unreachable("NEON::BI__builtin_neon_vaddlv_s8 NYI");
@@ -4340,9 +4345,6 @@ CIRGenFunction::emitAArch64BuiltinExpr(unsigned BuiltinID, const CallExpr *E,
43404345
case NEON::BI__builtin_neon_vaddlvq_s8: {
43414346
llvm_unreachable("NEON::BI__builtin_neon_vaddlvq_s8 NYI");
43424347
}
4343-
case NEON::BI__builtin_neon_vaddlvq_s16: {
4344-
llvm_unreachable("NEON::BI__builtin_neon_vaddlvq_s16 NYI");
4345-
}
43464348
case NEON::BI__builtin_neon_vsri_n_v:
43474349
case NEON::BI__builtin_neon_vsriq_n_v: {
43484350
llvm_unreachable("NEON::BI__builtin_neon_vsriq_n_v NYI");

clang/test/CIR/CodeGen/AArch64/neon-arith.c

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -894,6 +894,17 @@ uint32_t test_vaddlvq_u16(uint16x8_t a) {
894894
// LLVM: ret i32 [[VADDLV_I]]
895895
}
896896

897+
int32_t test_vaddlvq_s16(int16x8_t a) {
898+
return vaddlvq_s16(a);
899+
900+
// CIR-LABEL: vaddlvq_s16
901+
// CIR: cir.llvm.intrinsic "aarch64.neon.saddlv" {{%.*}}: (!cir.vector<!s16i x 8>) -> !s32i
902+
903+
// LLVM: {{.*}}test_vaddlvq_s16(<8 x i16>{{.*}}[[A:%.*]])
904+
// LLVM: [[VADDLV_I:%.*]] = call i32 @llvm.aarch64.neon.saddlv.i32.v8i16(<8 x i16> [[A]])
905+
// LLVM: ret i32 [[VADDLV_I]]
906+
}
907+
897908
uint16_t test_vaddv_u16(uint16x4_t a) {
898909
return vaddv_u16(a);
899910

0 commit comments

Comments
 (0)