================
@@ -2433,6 +2433,46 @@ static bool interp__builtin_elementwise_int_unaryop(
return true;
}
+static bool interp__builtin_elementwise_fp_binop(
+ InterpState &S, CodePtr OpPC, const CallExpr *Call,
+ llvm::function_ref<APFloat(const APFloat &, const APFloat &,
+ std::optional<APSInt> RoundingMode)>
+ Fn) {
+ assert((Call->getNumArgs() == 2) || (Call->getNumArgs() == 3));
+ const auto *VT = Call->getArg(0)->getType()->castAs<VectorType>();
+ assert(VT->getElementType()->isFloatingType());
+ unsigned NumElems = VT->getNumElements();
+
+ // Vector case.
+ assert(Call->getArg(0)->getType()->isVectorType() &&
+ Call->getArg(1)->getType()->isVectorType());
+ assert(VT->getElementType() ==
+ Call->getArg(1)->getType()->castAs<VectorType>()->getElementType());
+ assert(VT->getNumElements() ==
+ Call->getArg(1)->getType()->castAs<VectorType>()->getNumElements());
+
+ std::optional<APSInt> RoundingMode = std::nullopt;
+ if (Call->getNumArgs() == 3)
+ RoundingMode = popToAPSInt(S, Call->getArg(2));
+
+ const Pointer &BPtr = S.Stk.pop<Pointer>();
+ const Pointer &APtr = S.Stk.pop<Pointer>();
+ const Pointer &Dst = S.Stk.peek<Pointer>();
+ for (unsigned ElemIdx = 0; ElemIdx != NumElems; ++ElemIdx) {
+ using T = PrimConv<PT_Float>::T;
+ APFloat ElemA = APtr.elem<T>(ElemIdx).getAPFloat();
+ APFloat ElemB = BPtr.elem<T>(ElemIdx).getAPFloat();
+ if (ElemA.isNaN() || ElemA.isInfinity() || ElemA.isDenormal() ||
+ ElemB.isNaN() || ElemB.isInfinity() || ElemB.isDenormal())
+ return false;
----------------
RKSimon wrote:
I think eventually we're going to have to move this into the callback - and
possibly allow the callbacks to return false - but this should be enough for
min/max cases
https://github.com/llvm/llvm-project/pull/171966
_______________________________________________
cfe-commits mailing list
[email protected]
https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
