RPCS3/llvm-mirror
1
0
Fork 0
mirror of https://github.com/RPCS3/llvm-mirror.git synced 2024-11-22 02:33:06 +01:00
Code Issues Projects Releases Wiki Activity

[SelectionDAG] Add an overload of getStepVector that assumes step 1.

Browse Source 
This is mostly a minor convenience, but the pattern seems frequent
enough to be worthwhile (and we'll probably add more uses in the
future).

Differential Revision: https://reviews.llvm.org/D105850
This commit is contained in:
Eli Friedman 2021-07-12 15:11:01 -07:00
parent cf44692539
commit 0af449d2a7
4 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
include/llvm/CodeGen/SelectionDAG.h
View File

@ -837,6 +837,10 @@ public:
/// <0, Step, Step * 2, Step * 3, ...> /// <0, Step, Step * 2, Step * 3, ...>
SDValue getStepVector(const SDLoc &DL, EVT ResVT, SDValue Step); SDValue getStepVector(const SDLoc &DL, EVT ResVT, SDValue Step);
/// Returns a vector of type ResVT whose elements contain the linear sequence
/// <0, 1, 2, 3, ...>
SDValue getStepVector(const SDLoc &DL, EVT ResVT);
/// Returns an ISD::VECTOR_SHUFFLE node semantically equivalent to /// Returns an ISD::VECTOR_SHUFFLE node semantically equivalent to
/// the shuffle node in input but with swapped operands. /// the shuffle node in input but with swapped operands.
/// ///

5
lib/CodeGen/SelectionDAG/SelectionDAG.cpp
View File

@ -1747,6 +1747,11 @@ SDValue SelectionDAG::getCondCode(ISD::CondCode Cond) {
return SDValue(CondCodeNodes[Cond], 0); return SDValue(CondCodeNodes[Cond], 0);
} }
SDValue SelectionDAG::getStepVector(const SDLoc &DL, EVT ResVT) {
EVT OpVT = TLI->getTypeToTransformTo(*getContext(), ResVT.getScalarType());
return getStepVector(DL, ResVT, getConstant(1, DL, OpVT));
}
SDValue SelectionDAG::getStepVector(const SDLoc &DL, EVT ResVT, SDValue Step) { SDValue SelectionDAG::getStepVector(const SDLoc &DL, EVT ResVT, SDValue Step) {
if (ResVT.isScalableVector()) if (ResVT.isScalableVector())
return getNode(ISD::STEP_VECTOR, DL, ResVT, Step); return getNode(ISD::STEP_VECTOR, DL, ResVT, Step);

5
lib/CodeGen/SelectionDAG/SelectionDAGBuilder.cpp
View File

@ -11074,10 +11074,7 @@ void SelectionDAGBuilder::visitStepVector(const CallInst &I) {
const TargetLowering &TLI = DAG.getTargetLoweringInfo(); const TargetLowering &TLI = DAG.getTargetLoweringInfo();
auto DL = getCurSDLoc(); auto DL = getCurSDLoc();
EVT ResultVT = TLI.getValueType(DAG.getDataLayout(), I.getType()); EVT ResultVT = TLI.getValueType(DAG.getDataLayout(), I.getType());
EVT OpVT = setValue(&I, DAG.getStepVector(DL, ResultVT));
TLI.getTypeToTransformTo(*DAG.getContext(), ResultVT.getScalarType());
SDValue Step = DAG.getConstant(1, DL, OpVT);
setValue(&I, DAG.getStepVector(DL, ResultVT, Step));
} }
void SelectionDAGBuilder::visitVectorReverse(const CallInst &I) { void SelectionDAGBuilder::visitVectorReverse(const CallInst &I) {

6
lib/Target/AArch64/AArch64ISelLowering.cpp
View File

@ -9503,7 +9503,7 @@ SDValue AArch64TargetLowering::LowerDUPQLane(SDValue Op,
SDValue SplatOne = DAG.getNode(ISD::SPLAT_VECTOR, DL, MVT::nxv2i64, One); SDValue SplatOne = DAG.getNode(ISD::SPLAT_VECTOR, DL, MVT::nxv2i64, One);
// create the vector 0,1,0,1,... // create the vector 0,1,0,1,...
SDValue SV = DAG.getNode(ISD::STEP_VECTOR, DL, MVT::nxv2i64, One); SDValue SV = DAG.getStepVector(DL, MVT::nxv2i64);
SV = DAG.getNode(ISD::AND, DL, MVT::nxv2i64, SV, SplatOne); SV = DAG.getNode(ISD::AND, DL, MVT::nxv2i64, SV, SplatOne);
// create the vector idx64,idx64+1,idx64,idx64+1,... // create the vector idx64,idx64+1,idx64,idx64+1,...
@ -13998,9 +13998,7 @@ static SDValue LowerSVEIntrinsicIndex(SDNode *N, SelectionDAG &DAG) {
ScalarTy = MVT::i32; ScalarTy = MVT::i32;
// Lower index_vector(base, step) to mul(step step_vector(1)) + splat(base). // Lower index_vector(base, step) to mul(step step_vector(1)) + splat(base).
SDValue One = DAG.getConstant(1, DL, ScalarTy); SDValue StepVector = DAG.getStepVector(DL, N->getValueType(0));
SDValue StepVector =
DAG.getNode(ISD::STEP_VECTOR, DL, N->getValueType(0), One);
SDValue Step = DAG.getNode(ISD::SPLAT_VECTOR, DL, N->getValueType(0), Op2); SDValue Step = DAG.getNode(ISD::SPLAT_VECTOR, DL, N->getValueType(0), Op2);
SDValue Mul = DAG.getNode(ISD::MUL, DL, N->getValueType(0), StepVector, Step); SDValue Mul = DAG.getNode(ISD::MUL, DL, N->getValueType(0), StepVector, Step);
SDValue Base = DAG.getNode(ISD::SPLAT_VECTOR, DL, N->getValueType(0), Op1); SDValue Base = DAG.getNode(ISD::SPLAT_VECTOR, DL, N->getValueType(0), Op1);