diff --git a/llvm/include/llvm/Analysis/VectorUtils.h b/llvm/include/llvm/Analysis/VectorUtils.h --- a/llvm/include/llvm/Analysis/VectorUtils.h +++ b/llvm/include/llvm/Analysis/VectorUtils.h @@ -186,12 +186,13 @@ /// = "_LLVM_" /// = "N". Note: TLI does not support masked interfaces. /// = 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. /// = "v", as many as are the numArgs. /// = the name of the scalar function. /// = 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); diff --git a/llvm/lib/Analysis/VectorUtils.cpp b/llvm/lib/Analysis/VectorUtils.cpp --- a/llvm/lib/Analysis/VectorUtils.cpp +++ b/llvm/lib/Analysis/VectorUtils.cpp @@ -1300,10 +1300,14 @@ 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 << ")"; diff --git a/llvm/lib/Transforms/Utils/InjectTLIMappings.cpp b/llvm/lib/Transforms/Utils/InjectTLIMappings.cpp --- a/llvm/lib/Transforms/Utils/InjectTLIMappings.cpp +++ b/llvm/lib/Transforms/Utils/InjectTLIMappings.cpp @@ -96,7 +96,8 @@ 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; diff --git a/llvm/unittests/Analysis/VectorFunctionABITest.cpp b/llvm/unittests/Analysis/VectorFunctionABITest.cpp --- a/llvm/unittests/Analysis/VectorFunctionABITest.cpp +++ b/llvm/unittests/Analysis/VectorFunctionABITest.cpp @@ -101,9 +101,14 @@ // 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)"); }