void setPICLevel(PICLevel::Level PL);
/// @}
- /// @name Utility functions for querying and setting PGO counts
+ /// @name Utility functions for querying and setting PGO summary
/// @{
/// \brief Set maximum function count in PGO mode
/// \brief Returns maximum function count in PGO mode
Optional<uint64_t> getMaximumFunctionCount();
+
+ /// \brief Attach profile summary metadata to this module.
+ void setProfileSummary(Metadata *M);
+
+ /// \brief Returns profile summary metadata
+ Metadata *getProfileSummary();
/// @}
};
return None;
return cast<ConstantInt>(Val->getValue())->getZExtValue();
}
+
+void Module::setProfileSummary(Metadata *M) {
+ addModuleFlag(ModFlagBehavior::Error, "ProfileSummary", M);
+}
+
+Metadata *Module::getProfileSummary() {
+ return getModuleFlag("ProfileSummary");
+}
};
InstrProfSummary &PS = Reader->getSummary();
VerifySummary(PS);
+
+ // Test that conversion of summary to and from Metadata works.
Metadata *MD = PS.getMD(getGlobalContext());
ASSERT_TRUE(MD);
ProfileSummary *PSFromMD = ProfileSummary::getFromMD(MD);
InstrProfSummary *IPS = cast<InstrProfSummary>(PSFromMD);
VerifySummary(*IPS);
delete IPS;
+
+ // Test that summary can be attached to and read back from module.
+ Module M("my_module", getGlobalContext());
+ M.setProfileSummary(MD);
+ MD = M.getProfileSummary();
+ ASSERT_TRUE(MD);
+ PSFromMD = ProfileSummary::getFromMD(MD);
+ ASSERT_TRUE(PSFromMD);
+ ASSERT_TRUE(isa<InstrProfSummary>(PSFromMD));
+ IPS = cast<InstrProfSummary>(PSFromMD);
+ VerifySummary(*IPS);
+ delete IPS;
}
TEST_P(MaybeSparseInstrProfTest, get_icall_data_read_write) {
SampleProfileSummary &Summary = Reader->getSummary();
VerifySummary(Summary);
+ // Test that conversion of summary to and from Metadata works.
Metadata *MD = Summary.getMD(getGlobalContext());
ASSERT_TRUE(MD);
ProfileSummary *PS = ProfileSummary::getFromMD(MD);
SampleProfileSummary *SPS = cast<SampleProfileSummary>(PS);
VerifySummary(*SPS);
delete SPS;
+
+ // Test that summary can be attached to and read back from module.
+ Module M("my_module", getGlobalContext());
+ M.setProfileSummary(MD);
+ MD = M.getProfileSummary();
+ ASSERT_TRUE(MD);
+ PS = ProfileSummary::getFromMD(MD);
+ ASSERT_TRUE(PS);
+ ASSERT_TRUE(isa<SampleProfileSummary>(PS));
+ SPS = cast<SampleProfileSummary>(PS);
+ VerifySummary(*SPS);
+ delete SPS;
}
};