Skip to content

Commit aa02aa7

Browse files
ghehglanza
authored andcommitted
[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 7697fec commit aa02aa7

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
@@ -4330,11 +4330,16 @@ CIRGenFunction::emitAArch64BuiltinExpr(unsigned BuiltinID, const CallExpr *E,
43304330
case NEON::BI__builtin_neon_vaddlvq_u8: {
43314331
llvm_unreachable("NEON::BI__builtin_neon_vaddlvq_u8 NYI");
43324332
}
4333-
case NEON::BI__builtin_neon_vaddlvq_u16: {
4334-
mlir::Type argTy = cir::VectorType::get(builder.getContext(), UInt16Ty, 8);
4333+
case NEON::BI__builtin_neon_vaddlvq_u16:
4334+
usgn = true;
4335+
[[fallthrough]];
4336+
case NEON::BI__builtin_neon_vaddlvq_s16: {
4337+
mlir::Type argTy = cir::VectorType::get(builder.getContext(),
4338+
usgn ? UInt16Ty : SInt16Ty, 8);
43354339
llvm::SmallVector<mlir::Value, 1> argOps = {emitScalarExpr(E->getArg(0))};
4336-
return emitNeonCall(builder, {argTy}, argOps, "aarch64.neon.uaddlv",
4337-
UInt32Ty, getLoc(E->getExprLoc()));
4340+
return emitNeonCall(builder, {argTy}, argOps,
4341+
usgn ? "aarch64.neon.uaddlv" : "aarch64.neon.saddlv",
4342+
usgn ? UInt32Ty : SInt32Ty, getLoc(E->getExprLoc()));
43384343
}
43394344
case NEON::BI__builtin_neon_vaddlv_s8: {
43404345
llvm_unreachable("NEON::BI__builtin_neon_vaddlv_s8 NYI");
@@ -4345,9 +4350,6 @@ CIRGenFunction::emitAArch64BuiltinExpr(unsigned BuiltinID, const CallExpr *E,
43454350
case NEON::BI__builtin_neon_vaddlvq_s8: {
43464351
llvm_unreachable("NEON::BI__builtin_neon_vaddlvq_s8 NYI");
43474352
}
4348-
case NEON::BI__builtin_neon_vaddlvq_s16: {
4349-
llvm_unreachable("NEON::BI__builtin_neon_vaddlvq_s16 NYI");
4350-
}
43514353
case NEON::BI__builtin_neon_vsri_n_v:
43524354
case NEON::BI__builtin_neon_vsriq_n_v: {
43534355
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)