[clang] [Clang] VectorExprEvaluator::VisitCallExpr / InterpretBuiltin - add AVX512 VPTERNLOGD/VPTERNLOGQ intrinsics to be used in constexpr (PR #158703)
Timm Baeder via cfe-commits
cfe-commits at lists.llvm.org
Fri Oct 3 21:27:02 PDT 2025
================
@@ -2874,7 +2874,62 @@ static bool interp__builtin_x86_insert_subvector(InterpState &S, CodePtr OpPC,
});
Dst.initializeAllElements();
+ return true;
+}
+
+static bool interp__builtin_ia32_pternlog(InterpState &S, CodePtr OpPC,
+ const CallExpr *Call, bool MaskZ) {
+ assert(Call->getNumArgs() == 5);
+
+ const VectorType *VecT = Call->getArg(0)->getType()->castAs<VectorType>();
+ const PrimType &DstElemT = *S.getContext().classify(VecT->getElementType());
+ unsigned DstLen = VecT->getNumElements();
+ bool DstUnsigned =
+ VecT->getElementType()->isUnsignedIntegerOrEnumerationType();
+
+ APInt U = popToAPSInt(S, Call->getArg(4));
+ APInt Imm = popToAPSInt(S, Call->getArg(3));
+ const Pointer &C = S.Stk.pop<Pointer>();
+ const Pointer &B = S.Stk.pop<Pointer>();
+ const Pointer &A = S.Stk.pop<Pointer>();
+
+ const Pointer &Dst = S.Stk.peek<Pointer>();
+
+ for (unsigned I = 0; I < DstLen; ++I) {
+ APInt ALane;
+ APInt BLane;
+ APInt CLane;
+ INT_TYPE_SWITCH_NO_BOOL(DstElemT, {
+ ALane = A.elem<T>(I).toAPSInt();
+ BLane = B.elem<T>(I).toAPSInt();
+ CLane = C.elem<T>(I).toAPSInt();
+ });
+ const unsigned BitWidth = ALane.getBitWidth();
+ APInt RLane(BitWidth, 0);
+ if (U[I]) { // If lane not masked, compute ternary logic
+ for (unsigned Bit = 0; Bit < BitWidth; ++Bit) {
+ unsigned ABit = ALane[Bit];
+ unsigned BBit = BLane[Bit];
+ unsigned CBit = CLane[Bit];
+
+ unsigned Idx = (ABit << 2) | (BBit << 1) | (CBit);
+ RLane.setBitVal(Bit, Imm[Idx]);
+ }
+ INT_TYPE_SWITCH_NO_BOOL(DstElemT, {
+ Dst.elem<T>(I) = static_cast<T>(APSInt(RLane, DstUnsigned));
+ });
+ } else if (MaskZ) { // If zero masked, zero the lane
----------------
tbaederr wrote:
```suggestion
} else if (MaskZ) { // If zero masked, zero the lane.
```
https://github.com/llvm/llvm-project/pull/158703
More information about the cfe-commits
mailing list