Interface to get/set profile summary metadata to module

[eraman]

Summary

This is the first in a series of patches to remove MaxFunctionCount metadata from module and add profile summary metadata instead.

[davidxl]

Perhaps add a unit test in unittest/ProfileData/InstrProfTest.cpp?

[dnovillo]

Needs a test. Otherwise, LGTM. Thanks.

[eraman]

Added unit tests.

include/llvm/IR/Module.h

/// @{
 
   /// \brief Returns the PIC level (small or large model)
   PICLevel::Level getPICLevel() const;
 
   /// \brief Set the PIC level (small or large model)
   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
   void setMaximumFunctionCount(uint64_t);
 
   /// \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();
   /// @}
 };
 
 /// An raw_ostream inserter for modules.
 inline raw_ostream &operator<<(raw_ostream &O, const Module &M) {
   M.print(O, nullptr);
   return O;
 }

lib/IR/Module.cpp

 
 Optional<uint64_t> Module::getMaximumFunctionCount() {
   auto *Val =
       cast_or_null<ConstantAsMetadata>(getModuleFlag("MaxFunctionCount"));
   if (!Val)
     return None;
   return cast<ConstantInt>(Val->getValue())->getZExtValue();
 }
+
+void Module::setProfileSummary(Metadata *M) {
+  addModuleFlag(ModFlagBehavior::Error, "ProfileSummary", M);
+}
+
+Metadata *Module::getProfileSummary() {
+  return getModuleFlag("ProfileSummary");
+}