diff --git a/llvm/include/llvm/ProfileData/SampleProf.h b/llvm/include/llvm/ProfileData/SampleProf.h --- a/llvm/include/llvm/ProfileData/SampleProf.h +++ b/llvm/include/llvm/ProfileData/SampleProf.h @@ -195,19 +195,21 @@ /// The common profile is usually merged from profiles collected /// from running other targets. SecFlagPartial = (1 << 0), - /// SecFlagContext means this is context-sensitive profile for + /// SecFlagContext means this is context-sensitive flat profile for /// CSSPGO SecFlagFullContext = (1 << 1), /// SecFlagFSDiscriminator means this profile uses flow-sensitive /// discriminators. - SecFlagFSDiscriminator = (1 << 2) + SecFlagFSDiscriminator = (1 << 2), + /// SecFlagIsCSNested means this is context-sensitive nested profile for + /// CSSPGO + SecFlagIsCSNested = (1 << 4), }; enum class SecFuncMetadataFlags : uint32_t { SecFlagInvalid = 0, SecFlagIsProbeBased = (1 << 0), SecFlagHasAttribute = (1 << 1), - SecFlagIsCSNested = (1 << 2), }; enum class SecFuncOffsetFlags : uint32_t { diff --git a/llvm/lib/ProfileData/SampleProfReader.cpp b/llvm/lib/ProfileData/SampleProfReader.cpp --- a/llvm/lib/ProfileData/SampleProfReader.cpp +++ b/llvm/lib/ProfileData/SampleProfReader.cpp @@ -655,6 +655,8 @@ Summary->setPartialProfile(true); if (hasSecFlag(Entry, SecProfSummaryFlags::SecFlagFullContext)) FunctionSamples::ProfileIsCSFlat = ProfileIsCSFlat = true; + if (hasSecFlag(Entry, SecProfSummaryFlags::SecFlagIsCSNested)) + FunctionSamples::ProfileIsCSNested = ProfileIsCSNested; if (hasSecFlag(Entry, SecProfSummaryFlags::SecFlagFSDiscriminator)) FunctionSamples::ProfileIsFS = ProfileIsFS = true; break; @@ -688,9 +690,6 @@ ProfileIsProbeBased = hasSecFlag(Entry, SecFuncMetadataFlags::SecFlagIsProbeBased); FunctionSamples::ProfileIsProbeBased = ProfileIsProbeBased; - ProfileIsCSNested = - hasSecFlag(Entry, SecFuncMetadataFlags::SecFlagIsCSNested); - FunctionSamples::ProfileIsCSNested = ProfileIsCSNested; bool HasAttribute = hasSecFlag(Entry, SecFuncMetadataFlags::SecFlagHasAttribute); if (std::error_code EC = readFuncMetadata(HasAttribute)) @@ -1276,6 +1275,8 @@ Flags.append("partial,"); if (hasSecFlag(Entry, SecProfSummaryFlags::SecFlagFullContext)) Flags.append("context,"); + if (hasSecFlag(Entry, SecProfSummaryFlags::SecFlagIsCSNested)) + Flags.append("context-nested,"); if (hasSecFlag(Entry, SecProfSummaryFlags::SecFlagFSDiscriminator)) Flags.append("fs-discriminator,"); break; @@ -1288,8 +1289,6 @@ Flags.append("probe,"); if (hasSecFlag(Entry, SecFuncMetadataFlags::SecFlagHasAttribute)) Flags.append("attr,"); - if (hasSecFlag(Entry, SecFuncMetadataFlags::SecFlagIsCSNested)) - Flags.append("preinlined,"); break; default: break; diff --git a/llvm/lib/ProfileData/SampleProfWriter.cpp b/llvm/lib/ProfileData/SampleProfWriter.cpp --- a/llvm/lib/ProfileData/SampleProfWriter.cpp +++ b/llvm/lib/ProfileData/SampleProfWriter.cpp @@ -323,13 +323,13 @@ setToCompressSection(SecProfileSymbolList); if (Type == SecFuncMetadata && FunctionSamples::ProfileIsProbeBased) addSectionFlag(SecFuncMetadata, SecFuncMetadataFlags::SecFlagIsProbeBased); - if (Type == SecFuncMetadata && FunctionSamples::ProfileIsCSNested) - addSectionFlag(SecFuncMetadata, SecFuncMetadataFlags::SecFlagIsCSNested); if (Type == SecFuncMetadata && (FunctionSamples::ProfileIsCSFlat || FunctionSamples::ProfileIsCSNested)) addSectionFlag(SecFuncMetadata, SecFuncMetadataFlags::SecFlagHasAttribute); if (Type == SecProfSummary && FunctionSamples::ProfileIsCSFlat) addSectionFlag(SecProfSummary, SecProfSummaryFlags::SecFlagFullContext); + if (Type == SecProfSummary && FunctionSamples::ProfileIsCSNested) + addSectionFlag(SecProfSummary, SecProfSummaryFlags::SecFlagIsCSNested); if (Type == SecProfSummary && FunctionSamples::ProfileIsFS) addSectionFlag(SecProfSummary, SecProfSummaryFlags::SecFlagFSDiscriminator); diff --git a/llvm/test/tools/llvm-profdata/cs-sample-nested-profile.test b/llvm/test/tools/llvm-profdata/cs-sample-nested-profile.test --- a/llvm/test/tools/llvm-profdata/cs-sample-nested-profile.test +++ b/llvm/test/tools/llvm-profdata/cs-sample-nested-profile.test @@ -152,6 +152,4 @@ ; PROBE-NEXT: !CFGChecksum: 563022570642068 -; PREINLINE: FunctionMetadata {{.*}} Flags: {attr,preinlined} - - +; PREINLINE: ProfileSummarySection {{.*}} Flags: {context-nested}