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

[Analysis] Change VFABI::mangleTLIVectorName to use ElementCount

Browse Source 
Adds support for mangling TLI vector names for scalable vectors.

Differential Revision: https://reviews.llvm.org/D96338
This commit is contained in:
David Sherwood 2021-02-08 17:16:03 +00:00
parent 765390acdd
commit fa38810646
4 changed files with 19 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
include/llvm/Analysis/VectorUtils.h
View File

@ -186,12 +186,13 @@ Optional<VFInfo> tryDemangleForVFABI(StringRef MangledName, const Module &M);
/// <isa> = "_LLVM_"
/// <mask> = "N". Note: TLI does not support masked interfaces.
/// <vlen> = Number of concurrent lanes, stored in the `VectorizationFactor`
/// field of the `VecDesc` struct.
/// field of the `VecDesc` struct. If the number of lanes is scalable
/// then 'x' is printed instead.
/// <vparams> = "v", as many as are the numArgs.
/// <scalarname> = the name of the scalar function.
/// <vectorname> = the name of the vector function.
std::string mangleTLIVectorName(StringRef VectorName, StringRef ScalarName,
unsigned numArgs, unsigned VF);
unsigned numArgs, ElementCount VF);
/// Retrieve the `VFParamKind` from a string token.
VFParamKind getVFParamKindFromString(const StringRef Token);

8
lib/Analysis/VectorUtils.cpp
View File

@ -1300,10 +1300,14 @@ void InterleaveGroup<Instruction>::addMetadata(Instruction *NewInst) const {
std::string VFABI::mangleTLIVectorName(StringRef VectorName,
StringRef ScalarName, unsigned numArgs,
unsigned VF) {
ElementCount VF) {
SmallString<256> Buffer;
llvm::raw_svector_ostream Out(Buffer);
Out << "_ZGV" << VFABI::_LLVM_ << "N" << VF;
Out << "_ZGV" << VFABI::_LLVM_ << "N";
if (VF.isScalable())
Out << 'x';
else
Out << VF.getFixedValue();
for (unsigned I = 0; I < numArgs; ++I)
Out << "v";
Out << "_" << ScalarName << "(" << VectorName << ")";

3
lib/Transforms/Utils/InjectTLIMappings.cpp
View File

@ -96,7 +96,8 @@ static void addMappingsFromTLI(const TargetLibraryInfo &TLI, CallInst &CI) {
std::string(TLI.getVectorizedFunction(ScalarName, VF));
if (!TLIName.empty()) {
std::string MangledName = VFABI::mangleTLIVectorName(
TLIName, ScalarName, CI.getNumArgOperands(), VF);
TLIName, ScalarName, CI.getNumArgOperands(),
ElementCount::getFixed(VF));
if (!OriginalSetOfMappings.count(MangledName)) {
Mappings.push_back(MangledName);
++NumCallInjected;

11
unittests/Analysis/VectorFunctionABITest.cpp
View File

@ -101,9 +101,14 @@ protected:
// This test makes sure correct mangling occurs for given string.
TEST_F(VFABIParserTest, ManglingVectorTLINames) {
EXPECT_EQ(VFABI::mangleTLIVectorName("vec", "scalar", 3, 4),
"_ZGV_LLVM_N4vvv_scalar(vec)");
EXPECT_EQ(VFABI::mangleTLIVectorName("custom.call.v5", "custom.call", 1, 5),
EXPECT_EQ(
VFABI::mangleTLIVectorName("vec", "scalar", 3, ElementCount::getFixed(4)),
"_ZGV_LLVM_N4vvv_scalar(vec)");
EXPECT_EQ(VFABI::mangleTLIVectorName("vec", "scalar", 3,
ElementCount::getScalable(4)),
"_ZGV_LLVM_Nxvvv_scalar(vec)");
EXPECT_EQ(VFABI::mangleTLIVectorName("custom.call.v5", "custom.call", 1,
ElementCount::getFixed(5)),
"_ZGV_LLVM_N5v_custom.call(custom.call.v5)");
}