[llvm] [AArch64][SME] Allow memory operations lowering to custom SME functions. (PR #79263)
Sander de Smalen via llvm-commits
llvm-commits at lists.llvm.org
Mon Feb 5 08:47:05 PST 2024
================
@@ -76,12 +82,68 @@ SDValue AArch64SelectionDAGInfo::EmitMOPS(AArch64ISD::NodeType SDOpcode,
}
}
+SDValue AArch64SelectionDAGInfo::EmitSpecializedLibcall(
+ SelectionDAG &DAG, const SDLoc &DL, SDValue Chain, SDValue Dst, SDValue Src,
+ SDValue Size, RTLIB::Libcall LC) const {
+ const AArch64Subtarget &STI =
+ DAG.getMachineFunction().getSubtarget<AArch64Subtarget>();
+ const AArch64TargetLowering *TLI = STI.getTargetLowering();
+ TargetLowering::ArgListTy Args;
+ TargetLowering::ArgListEntry Entry;
+ SDValue Symbol;
+ Entry.Ty = DAG.getDataLayout().getIntPtrType(*DAG.getContext());
+ Entry.Node = Dst;
+ Args.push_back(Entry);
+ EVT Ty = TLI->getPointerTy(DAG.getDataLayout());
+
+ if (!EnableSMEMops)
+ return SDValue();
+
+ switch (LC) {
+ case RTLIB::MEMCPY:
+ Symbol = DAG.getExternalSymbol("__arm_sc_memcpy", Ty);
+ Entry.Node = Src;
+ Args.push_back(Entry);
+ break;
+ case RTLIB::MEMMOVE:
+ Symbol = DAG.getExternalSymbol("__arm_sc_memmove", Ty);
+ Entry.Node = Src;
+ Args.push_back(Entry);
+ break;
+ case RTLIB::MEMSET:
+ Symbol = DAG.getExternalSymbol("__arm_sc_memset", Ty);
+ Src = DAG.getZExtOrTrunc(Src, DL, MVT::i32);
+ Entry.Node = Src;
+ Entry.Ty = Type::getInt32Ty(*DAG.getContext());
+ Args.push_back(Entry);
+ break;
+ default:
+ return SDValue();
+ }
+ Entry.Node = Size;
+ Entry.Ty = DAG.getDataLayout().getIntPtrType(*DAG.getContext());
+ Args.push_back(Entry);
+
+ TargetLowering::CallLoweringInfo CLI(DAG);
+ CLI.setDebugLoc(DL).setChain(Chain).setLibCallee(
+ TLI->getLibcallCallingConv(RTLIB::MEMCPY),
----------------
sdesmalen-arm wrote:
```suggestion
TLI->getLibcallCallingConv(LC),
```
How did this get changed back?
https://github.com/llvm/llvm-project/pull/79263
More information about the llvm-commits
mailing list