================
@@ -2773,6 +2773,62 @@ static bool interp__builtin_blend(InterpState &S,
CodePtr OpPC,
return true;
}
+enum class Half { None, Low, High };
+
+static bool interp__builtin_ia32_pshuf(InterpState &S, CodePtr OpPC,
+ const CallExpr *Call, Half whichHalf) {
+ assert(Call->getNumArgs() == 2 && "masked forms handled via select*");
+ APSInt ControlImm = popToAPSInt(S, Call->getArg(1));
+ const Pointer &Src = S.Stk.pop<Pointer>();
+ const Pointer &Dst = S.Stk.peek<Pointer>();
+
+ unsigned NumElems = Dst.getNumElems();
+ PrimType ElemT = Dst.getFieldDesc()->getPrimType();
+
+ unsigned ElemBits = static_cast<unsigned>(primSize(ElemT) * 8);
+ if (ElemBits != 16 && ElemBits != 32)
+ return false;
+
+ unsigned LaneElts = 128u / ElemBits;
+ assert(LaneElts && (NumElems % LaneElts == 0));
+
+ uint8_t ctl = static_cast<uint8_t>(ControlImm.getZExtValue());
+
+ for (unsigned idx = 0; idx != NumElems; idx++) {
+ unsigned LaneBase = (idx / LaneElts) * LaneElts;
+ unsigned LaneIdx = idx % LaneElts;
+
+ unsigned SrcIdx = idx;
+
+ if (ElemBits == 32) {
+ unsigned sel = (ctl >> (2 * LaneIdx)) & 0x3;
+ SrcIdx = LaneBase + sel;
+ } else {
+ if (LaneElts == 4) {
+ unsigned sel = (ctl >> (2 * LaneIdx)) & 0x3;
+ SrcIdx = LaneBase + sel;
+ } else {
+ constexpr unsigned HalfSize = 4;
+ if (whichHalf == Half::Low && LaneIdx < HalfSize) {
+ unsigned sel = (ctl >> (2 * LaneIdx)) & 0x3;
+ SrcIdx = LaneBase + sel;
+ } else if (whichHalf == Half::High && LaneIdx >= HalfSize) {
+ unsigned rel = LaneIdx - HalfSize;
+ unsigned sel = (ctl >> (2 * rel)) & 0x3;
+ SrcIdx = LaneBase + HalfSize + sel;
+ } else if (whichHalf == Half::None) {
+ unsigned sel = (ctl >> (2 * LaneIdx)) & 0x3;
----------------
RKSimon wrote:
Can this case even occur? Doesn't LaneElts == 4 catch it?
https://github.com/llvm/llvm-project/pull/161210
_______________________________________________
cfe-commits mailing list
[email protected]
https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
