include/llvm/Analysis/CGSCCPassManager.h

   1 //===- CGSCCPassManager.h - Call graph pass management ----------*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 /// \file
  10 ///
  11 /// This header provides classes for managing passes over SCCs of the call
  12 /// graph. These passes form an important component of LLVM's interprocedural
  13 /// optimizations. Because they operate on the SCCs of the call graph, and they
  14 /// wtraverse the graph in post order, they can effectively do pair-wise
  15 /// interprocedural optimizations for all call edges in the program. At each
  16 /// call site edge, the callee has already been optimized as much as is
  17 /// possible. This in turn allows very accurate analysis of it for IPO.
  18 ///
  19 //===----------------------------------------------------------------------===//
  20
  21 #ifndef LLVM_ANALYSIS_CGSCCPASSMANAGER_H
  22 #define LLVM_ANALYSIS_CGSCCPASSMANAGER_H
  23
  24 #include "llvm/IR/PassManager.h"
  25 #include "llvm/Analysis/LazyCallGraph.h"
  26
  27 namespace llvm {
  28
  29 class CGSCCAnalysisManager;
  30
  31 class CGSCCPassManager {
  32 public:
  33   // We have to explicitly define all the special member functions because MSVC
  34   // refuses to generate them.
  35   CGSCCPassManager() {}
  36   CGSCCPassManager(CGSCCPassManager &&Arg) : Passes(std::move(Arg.Passes)) {}
  37   CGSCCPassManager &operator=(CGSCCPassManager &&RHS) {
  38     Passes = std::move(RHS.Passes);
  39     return *this;
  40   }
  41
  42   /// \brief Run all of the CGSCC passes in this pass manager over a SCC.
  43   PreservedAnalyses run(LazyCallGraph::SCC &C,
  44                         CGSCCAnalysisManager *AM = nullptr);
  45
  46   template <typename CGSCCPassT> void addPass(CGSCCPassT Pass) {
  47     Passes.emplace_back(new CGSCCPassModel<CGSCCPassT>(std::move(Pass)));
  48   }
  49
  50   static StringRef name() { return "CGSCCPassManager"; }
  51
  52 private:
  53   // Pull in the concept type and model template specialized for SCCs.
  54   typedef detail::PassConcept<LazyCallGraph::SCC &, CGSCCAnalysisManager>
  55   CGSCCPassConcept;
  56   template <typename PassT>
  57   struct CGSCCPassModel
  58       : detail::PassModel<LazyCallGraph::SCC &, CGSCCAnalysisManager, PassT> {
  59     CGSCCPassModel(PassT Pass)
  60         : detail::PassModel<LazyCallGraph::SCC &, CGSCCAnalysisManager, PassT>(
  61               std::move(Pass)) {}
  62   };
  63
  64   CGSCCPassManager(const CGSCCPassManager &) LLVM_DELETED_FUNCTION;
  65   CGSCCPassManager &operator=(const CGSCCPassManager &) LLVM_DELETED_FUNCTION;
  66
  67   std::vector<std::unique_ptr<CGSCCPassConcept>> Passes;
  68 };
  69
  70 /// \brief A function analysis manager to coordinate and cache analyses run over
  71 /// a module.
  72 class CGSCCAnalysisManager : public detail::AnalysisManagerBase<
  73                                  CGSCCAnalysisManager, LazyCallGraph::SCC &> {
  74   friend class detail::AnalysisManagerBase<CGSCCAnalysisManager,
  75                                            LazyCallGraph::SCC &>;
  76   typedef detail::AnalysisManagerBase<CGSCCAnalysisManager,
  77                                       LazyCallGraph::SCC &> BaseT;
  78   typedef BaseT::ResultConceptT ResultConceptT;
  79   typedef BaseT::PassConceptT PassConceptT;
  80
  81 public:
  82   // Most public APIs are inherited from the CRTP base class.
  83
  84   // We have to explicitly define all the special member functions because MSVC
  85   // refuses to generate them.
  86   CGSCCAnalysisManager() {}
  87   CGSCCAnalysisManager(CGSCCAnalysisManager &&Arg)
  88       : BaseT(std::move(static_cast<BaseT &>(Arg))),
  89         CGSCCAnalysisResults(std::move(Arg.CGSCCAnalysisResults)) {}
  90   CGSCCAnalysisManager &operator=(CGSCCAnalysisManager &&RHS) {
  91     BaseT::operator=(std::move(static_cast<BaseT &>(RHS)));
  92     CGSCCAnalysisResults = std::move(RHS.CGSCCAnalysisResults);
  93     return *this;
  94   }
  95
  96   /// \brief Returns true if the analysis manager has an empty results cache.
  97   bool empty() const;
  98
  99   /// \brief Clear the function analysis result cache.
 100   ///
 101   /// This routine allows cleaning up when the set of functions itself has
 102   /// potentially changed, and thus we can't even look up a a result and
 103   /// invalidate it directly. Notably, this does *not* call invalidate
 104   /// functions as there is nothing to be done for them.
 105   void clear();
 106
 107 private:
 108   CGSCCAnalysisManager(const CGSCCAnalysisManager &) LLVM_DELETED_FUNCTION;
 109   CGSCCAnalysisManager &
 110   operator=(const CGSCCAnalysisManager &) LLVM_DELETED_FUNCTION;
 111
 112   /// \brief Get a function pass result, running the pass if necessary.
 113   ResultConceptT &getResultImpl(void *PassID, LazyCallGraph::SCC &C);
 114
 115   /// \brief Get a cached function pass result or return null.
 116   ResultConceptT *getCachedResultImpl(void *PassID,
 117                                       LazyCallGraph::SCC &C) const;
 118
 119   /// \brief Invalidate a function pass result.
 120   void invalidateImpl(void *PassID, LazyCallGraph::SCC &C);
 121
 122   /// \brief Invalidate the results for a function..
 123   void invalidateImpl(LazyCallGraph::SCC &C, const PreservedAnalyses &PA);
 124
 125   /// \brief List of function analysis pass IDs and associated concept pointers.
 126   ///
 127   /// Requires iterators to be valid across appending new entries and arbitrary
 128   /// erases. Provides both the pass ID and concept pointer such that it is
 129   /// half of a bijection and provides storage for the actual result concept.
 130   typedef std::list<
 131       std::pair<void *, std::unique_ptr<detail::AnalysisResultConcept<
 132                             LazyCallGraph::SCC &>>>> CGSCCAnalysisResultListT;
 133
 134   /// \brief Map type from function pointer to our custom list type.
 135   typedef DenseMap<LazyCallGraph::SCC *, CGSCCAnalysisResultListT>
 136   CGSCCAnalysisResultListMapT;
 137
 138   /// \brief Map from function to a list of function analysis results.
 139   ///
 140   /// Provides linear time removal of all analysis results for a function and
 141   /// the ultimate storage for a particular cached analysis result.
 142   CGSCCAnalysisResultListMapT CGSCCAnalysisResultLists;
 143
 144   /// \brief Map type from a pair of analysis ID and function pointer to an
 145   /// iterator into a particular result list.
 146   typedef DenseMap<std::pair<void *, LazyCallGraph::SCC *>,
 147                    CGSCCAnalysisResultListT::iterator> CGSCCAnalysisResultMapT;
 148
 149   /// \brief Map from an analysis ID and function to a particular cached
 150   /// analysis result.
 151   CGSCCAnalysisResultMapT CGSCCAnalysisResults;
 152 };
 153
 154 /// \brief A module analysis which acts as a proxy for a CGSCC analysis
 155 /// manager.
 156 ///
 157 /// This primarily proxies invalidation information from the module analysis
 158 /// manager and module pass manager to a CGSCC analysis manager. You should
 159 /// never use a CGSCC analysis manager from within (transitively) a module
 160 /// pass manager unless your parent module pass has received a proxy result
 161 /// object for it.
 162 class CGSCCAnalysisManagerModuleProxy {
 163 public:
 164   class Result {
 165   public:
 166     explicit Result(CGSCCAnalysisManager &CGAM) : CGAM(&CGAM) {}
 167     // We have to explicitly define all the special member functions because
 168     // MSVC refuses to generate them.
 169     Result(const Result &Arg) : CGAM(Arg.CGAM) {}
 170     Result(Result &&Arg) : CGAM(std::move(Arg.CGAM)) {}
 171     Result &operator=(Result RHS) {
 172       std::swap(CGAM, RHS.CGAM);
 173       return *this;
 174     }
 175     ~Result();
 176
 177     /// \brief Accessor for the \c CGSCCAnalysisManager.
 178     CGSCCAnalysisManager &getManager() { return *CGAM; }
 179
 180     /// \brief Handler for invalidation of the module.
 181     ///
 182     /// If this analysis itself is preserved, then we assume that the call
 183     /// graph of the module hasn't changed and thus we don't need to invalidate
 184     /// *all* cached data associated with a \c SCC* in the \c
 185     /// CGSCCAnalysisManager.
 186     ///
 187     /// Regardless of whether this analysis is marked as preserved, all of the
 188     /// analyses in the \c CGSCCAnalysisManager are potentially invalidated
 189     /// based on the set of preserved analyses.
 190     bool invalidate(Module &M, const PreservedAnalyses &PA);
 191
 192   private:
 193     CGSCCAnalysisManager *CGAM;
 194   };
 195
 196   static void *ID() { return (void *)&PassID; }
 197
 198   static StringRef name() { return "CGSCCAnalysisManagerModuleProxy"; }
 199
 200   explicit CGSCCAnalysisManagerModuleProxy(CGSCCAnalysisManager &CGAM)
 201       : CGAM(&CGAM) {}
 202   // We have to explicitly define all the special member functions because MSVC
 203   // refuses to generate them.
 204   CGSCCAnalysisManagerModuleProxy(
 205       const CGSCCAnalysisManagerModuleProxy &Arg)
 206       : CGAM(Arg.CGAM) {}
 207   CGSCCAnalysisManagerModuleProxy(CGSCCAnalysisManagerModuleProxy &&Arg)
 208       : CGAM(std::move(Arg.CGAM)) {}
 209   CGSCCAnalysisManagerModuleProxy &
 210   operator=(CGSCCAnalysisManagerModuleProxy RHS) {
 211     std::swap(CGAM, RHS.CGAM);
 212     return *this;
 213   }
 214
 215   /// \brief Run the analysis pass and create our proxy result object.
 216   ///
 217   /// This doesn't do any interesting work, it is primarily used to insert our
 218   /// proxy result object into the module analysis cache so that we can proxy
 219   /// invalidation to the CGSCC analysis manager.
 220   ///
 221   /// In debug builds, it will also assert that the analysis manager is empty
 222   /// as no queries should arrive at the CGSCC analysis manager prior to
 223   /// this analysis being requested.
 224   Result run(Module &M);
 225
 226 private:
 227   static char PassID;
 228
 229   CGSCCAnalysisManager *CGAM;
 230 };
 231
 232 /// \brief A CGSCC analysis which acts as a proxy for a module analysis
 233 /// manager.
 234 ///
 235 /// This primarily provides an accessor to a parent module analysis manager to
 236 /// CGSCC passes. Only the const interface of the module analysis manager is
 237 /// provided to indicate that once inside of a CGSCC analysis pass you
 238 /// cannot request a module analysis to actually run. Instead, the user must
 239 /// rely on the \c getCachedResult API.
 240 ///
 241 /// This proxy *doesn't* manage the invalidation in any way. That is handled by
 242 /// the recursive return path of each layer of the pass manager and the
 243 /// returned PreservedAnalysis set.
 244 class ModuleAnalysisManagerCGSCCProxy {
 245 public:
 246   /// \brief Result proxy object for \c ModuleAnalysisManagerCGSCCProxy.
 247   class Result {
 248   public:
 249     explicit Result(const ModuleAnalysisManager &MAM) : MAM(&MAM) {}
 250     // We have to explicitly define all the special member functions because
 251     // MSVC refuses to generate them.
 252     Result(const Result &Arg) : MAM(Arg.MAM) {}
 253     Result(Result &&Arg) : MAM(std::move(Arg.MAM)) {}
 254     Result &operator=(Result RHS) {
 255       std::swap(MAM, RHS.MAM);
 256       return *this;
 257     }
 258
 259     const ModuleAnalysisManager &getManager() const { return *MAM; }
 260
 261     /// \brief Handle invalidation by ignoring it, this pass is immutable.
 262     bool invalidate(LazyCallGraph::SCC &) { return false; }
 263
 264   private:
 265     const ModuleAnalysisManager *MAM;
 266   };
 267
 268   static void *ID() { return (void *)&PassID; }
 269
 270   static StringRef name() { return "ModuleAnalysisManagerCGSCCProxy"; }
 271
 272   ModuleAnalysisManagerCGSCCProxy(const ModuleAnalysisManager &MAM)
 273       : MAM(&MAM) {}
 274   // We have to explicitly define all the special member functions because MSVC
 275   // refuses to generate them.
 276   ModuleAnalysisManagerCGSCCProxy(
 277       const ModuleAnalysisManagerCGSCCProxy &Arg)
 278       : MAM(Arg.MAM) {}
 279   ModuleAnalysisManagerCGSCCProxy(ModuleAnalysisManagerCGSCCProxy &&Arg)
 280       : MAM(std::move(Arg.MAM)) {}
 281   ModuleAnalysisManagerCGSCCProxy &
 282   operator=(ModuleAnalysisManagerCGSCCProxy RHS) {
 283     std::swap(MAM, RHS.MAM);
 284     return *this;
 285   }
 286
 287   /// \brief Run the analysis pass and create our proxy result object.
 288   /// Nothing to see here, it just forwards the \c MAM reference into the
 289   /// result.
 290   Result run(LazyCallGraph::SCC &) { return Result(*MAM); }
 291
 292 private:
 293   static char PassID;
 294
 295   const ModuleAnalysisManager *MAM;
 296 };
 297
 298 /// \brief The core module pass which does a post-order walk of the SCCs and
 299 /// runs a CGSCC pass over each one.
 300 ///
 301 /// Designed to allow composition of a CGSCCPass(Manager) and
 302 /// a ModulePassManager. Note that this pass must be run with a module analysis
 303 /// manager as it uses the LazyCallGraph analysis. It will also run the
 304 /// \c CGSCCAnalysisManagerModuleProxy analysis prior to running the CGSCC
 305 /// pass over the module to enable a \c FunctionAnalysisManager to be used
 306 /// within this run safely.
 307 template <typename CGSCCPassT> class ModuleToPostOrderCGSCCPassAdaptor {
 308 public:
 309   explicit ModuleToPostOrderCGSCCPassAdaptor(CGSCCPassT Pass)
 310       : Pass(std::move(Pass)) {}
 311   // We have to explicitly define all the special member functions because MSVC
 312   // refuses to generate them.
 313   ModuleToPostOrderCGSCCPassAdaptor(
 314       const ModuleToPostOrderCGSCCPassAdaptor &Arg)
 315       : Pass(Arg.Pass) {}
 316   ModuleToPostOrderCGSCCPassAdaptor(ModuleToPostOrderCGSCCPassAdaptor &&Arg)
 317       : Pass(std::move(Arg.Pass)) {}
 318   friend void swap(ModuleToPostOrderCGSCCPassAdaptor &LHS,
 319                    ModuleToPostOrderCGSCCPassAdaptor &RHS) {
 320     using std::swap;
 321     swap(LHS.Pass, RHS.Pass);
 322   }
 323   ModuleToPostOrderCGSCCPassAdaptor &
 324   operator=(ModuleToPostOrderCGSCCPassAdaptor RHS) {
 325     swap(*this, RHS);
 326     return *this;
 327   }
 328
 329   /// \brief Runs the CGSCC pass across every SCC in the module.
 330   PreservedAnalyses run(Module &M, ModuleAnalysisManager *AM) {
 331     assert(AM && "We need analyses to compute the call graph!");
 332
 333     // Setup the CGSCC analysis manager from its proxy.
 334     CGSCCAnalysisManager &CGAM =
 335         AM->getResult<CGSCCAnalysisManagerModuleProxy>(M).getManager();
 336
 337     // Get the call graph for this module.
 338     LazyCallGraph &CG = AM->getResult<LazyCallGraphAnalysis>(M);
 339
 340     PreservedAnalyses PA = PreservedAnalyses::all();
 341     for (LazyCallGraph::SCC &C : CG.postorder_sccs()) {
 342       PreservedAnalyses PassPA = Pass.run(C, &CGAM);
 343
 344       // We know that the CGSCC pass couldn't have invalidated any other
 345       // SCC's analyses (that's the contract of a CGSCC pass), so
 346       // directly handle the CGSCC analysis manager's invalidation here.
 347       // FIXME: This isn't quite correct. We need to handle the case where the
 348       // pass updated the CG, particularly some child of the current SCC, and
 349       // invalidate its analyses.
 350       CGAM.invalidate(C, PassPA);
 351
 352       // Then intersect the preserved set so that invalidation of module
 353       // analyses will eventually occur when the module pass completes.
 354       PA.intersect(std::move(PassPA));
 355     }
 356
 357     // By definition we preserve the proxy. This precludes *any* invalidation
 358     // of CGSCC analyses by the proxy, but that's OK because we've taken
 359     // care to invalidate analyses in the CGSCC analysis manager
 360     // incrementally above.
 361     PA.preserve<CGSCCAnalysisManagerModuleProxy>();
 362     return PA;
 363   }
 364
 365   static StringRef name() { return "ModuleToPostOrderCGSCCPassAdaptor"; }
 366
 367 private:
 368   CGSCCPassT Pass;
 369 };
 370
 371 /// \brief A function to deduce a function pass type and wrap it in the
 372 /// templated adaptor.
 373 template <typename CGSCCPassT>
 374 ModuleToPostOrderCGSCCPassAdaptor<CGSCCPassT>
 375 createModuleToPostOrderCGSCCPassAdaptor(CGSCCPassT Pass) {
 376   return std::move(
 377       ModuleToPostOrderCGSCCPassAdaptor<CGSCCPassT>(std::move(Pass)));
 378 }
 379
 380 /// \brief A CGSCC analysis which acts as a proxy for a function analysis
 381 /// manager.
 382 ///
 383 /// This primarily proxies invalidation information from the CGSCC analysis
 384 /// manager and CGSCC pass manager to a function analysis manager. You should
 385 /// never use a function analysis manager from within (transitively) a CGSCC
 386 /// pass manager unless your parent CGSCC pass has received a proxy result
 387 /// object for it.
 388 class FunctionAnalysisManagerCGSCCProxy {
 389 public:
 390   class Result {
 391   public:
 392     explicit Result(FunctionAnalysisManager &FAM) : FAM(&FAM) {}
 393     // We have to explicitly define all the special member functions because
 394     // MSVC refuses to generate them.
 395     Result(const Result &Arg) : FAM(Arg.FAM) {}
 396     Result(Result &&Arg) : FAM(std::move(Arg.FAM)) {}
 397     Result &operator=(Result RHS) {
 398       std::swap(FAM, RHS.FAM);
 399       return *this;
 400     }
 401     ~Result();
 402
 403     /// \brief Accessor for the \c FunctionAnalysisManager.
 404     FunctionAnalysisManager &getManager() { return *FAM; }
 405
 406     /// \brief Handler for invalidation of the SCC.
 407     ///
 408     /// If this analysis itself is preserved, then we assume that the set of \c
 409     /// Function objects in the \c SCC hasn't changed and thus we don't need
 410     /// to invalidate *all* cached data associated with a \c Function* in the \c
 411     /// FunctionAnalysisManager.
 412     ///
 413     /// Regardless of whether this analysis is marked as preserved, all of the
 414     /// analyses in the \c FunctionAnalysisManager are potentially invalidated
 415     /// based on the set of preserved analyses.
 416     bool invalidate(LazyCallGraph::SCC &C, const PreservedAnalyses &PA);
 417
 418   private:
 419     FunctionAnalysisManager *FAM;
 420   };
 421
 422   static void *ID() { return (void *)&PassID; }
 423
 424   static StringRef name() { return "FunctionAnalysisManagerCGSCCProxy"; }
 425
 426   explicit FunctionAnalysisManagerCGSCCProxy(FunctionAnalysisManager &FAM)
 427       : FAM(&FAM) {}
 428   // We have to explicitly define all the special member functions because MSVC
 429   // refuses to generate them.
 430   FunctionAnalysisManagerCGSCCProxy(
 431       const FunctionAnalysisManagerCGSCCProxy &Arg)
 432       : FAM(Arg.FAM) {}
 433   FunctionAnalysisManagerCGSCCProxy(FunctionAnalysisManagerCGSCCProxy &&Arg)
 434       : FAM(std::move(Arg.FAM)) {}
 435   FunctionAnalysisManagerCGSCCProxy &
 436   operator=(FunctionAnalysisManagerCGSCCProxy RHS) {
 437     std::swap(FAM, RHS.FAM);
 438     return *this;
 439   }
 440
 441   /// \brief Run the analysis pass and create our proxy result object.
 442   ///
 443   /// This doesn't do any interesting work, it is primarily used to insert our
 444   /// proxy result object into the module analysis cache so that we can proxy
 445   /// invalidation to the function analysis manager.
 446   ///
 447   /// In debug builds, it will also assert that the analysis manager is empty
 448   /// as no queries should arrive at the function analysis manager prior to
 449   /// this analysis being requested.
 450   Result run(LazyCallGraph::SCC &C);
 451
 452 private:
 453   static char PassID;
 454
 455   FunctionAnalysisManager *FAM;
 456 };
 457
 458 /// \brief A function analysis which acts as a proxy for a CGSCC analysis
 459 /// manager.
 460 ///
 461 /// This primarily provides an accessor to a parent CGSCC analysis manager to
 462 /// function passes. Only the const interface of the CGSCC analysis manager is
 463 /// provided to indicate that once inside of a function analysis pass you
 464 /// cannot request a CGSCC analysis to actually run. Instead, the user must
 465 /// rely on the \c getCachedResult API.
 466 ///
 467 /// This proxy *doesn't* manage the invalidation in any way. That is handled by
 468 /// the recursive return path of each layer of the pass manager and the
 469 /// returned PreservedAnalysis set.
 470 class CGSCCAnalysisManagerFunctionProxy {
 471 public:
 472   /// \brief Result proxy object for \c ModuleAnalysisManagerFunctionProxy.
 473   class Result {
 474   public:
 475     explicit Result(const CGSCCAnalysisManager &CGAM) : CGAM(&CGAM) {}
 476     // We have to explicitly define all the special member functions because
 477     // MSVC refuses to generate them.
 478     Result(const Result &Arg) : CGAM(Arg.CGAM) {}
 479     Result(Result &&Arg) : CGAM(std::move(Arg.CGAM)) {}
 480     Result &operator=(Result RHS) {
 481       std::swap(CGAM, RHS.CGAM);
 482       return *this;
 483     }
 484
 485     const CGSCCAnalysisManager &getManager() const { return *CGAM; }
 486
 487     /// \brief Handle invalidation by ignoring it, this pass is immutable.
 488     bool invalidate(Function &) { return false; }
 489
 490   private:
 491     const CGSCCAnalysisManager *CGAM;
 492   };
 493
 494   static void *ID() { return (void *)&PassID; }
 495
 496   static StringRef name() { return "CGSCCAnalysisManagerFunctionProxy"; }
 497
 498   CGSCCAnalysisManagerFunctionProxy(const CGSCCAnalysisManager &CGAM)
 499       : CGAM(&CGAM) {}
 500   // We have to explicitly define all the special member functions because MSVC
 501   // refuses to generate them.
 502   CGSCCAnalysisManagerFunctionProxy(
 503       const CGSCCAnalysisManagerFunctionProxy &Arg)
 504       : CGAM(Arg.CGAM) {}
 505   CGSCCAnalysisManagerFunctionProxy(CGSCCAnalysisManagerFunctionProxy &&Arg)
 506       : CGAM(std::move(Arg.CGAM)) {}
 507   CGSCCAnalysisManagerFunctionProxy &
 508   operator=(CGSCCAnalysisManagerFunctionProxy RHS) {
 509     std::swap(CGAM, RHS.CGAM);
 510     return *this;
 511   }
 512
 513   /// \brief Run the analysis pass and create our proxy result object.
 514   /// Nothing to see here, it just forwards the \c CGAM reference into the
 515   /// result.
 516   Result run(Function &) { return Result(*CGAM); }
 517
 518 private:
 519   static char PassID;
 520
 521   const CGSCCAnalysisManager *CGAM;
 522 };
 523
 524 /// \brief Adaptor that maps from a SCC to its functions.
 525 ///
 526 /// Designed to allow composition of a FunctionPass(Manager) and
 527 /// a CGSCCPassManager. Note that if this pass is constructed with a pointer
 528 /// to a \c CGSCCAnalysisManager it will run the
 529 /// \c FunctionAnalysisManagerCGSCCProxy analysis prior to running the function
 530 /// pass over the SCC to enable a \c FunctionAnalysisManager to be used
 531 /// within this run safely.
 532 template <typename FunctionPassT> class CGSCCToFunctionPassAdaptor {
 533 public:
 534   explicit CGSCCToFunctionPassAdaptor(FunctionPassT Pass)
 535       : Pass(std::move(Pass)) {}
 536   // We have to explicitly define all the special member functions because MSVC
 537   // refuses to generate them.
 538   CGSCCToFunctionPassAdaptor(const CGSCCToFunctionPassAdaptor &Arg)
 539       : Pass(Arg.Pass) {}
 540   CGSCCToFunctionPassAdaptor(CGSCCToFunctionPassAdaptor &&Arg)
 541       : Pass(std::move(Arg.Pass)) {}
 542   friend void swap(CGSCCToFunctionPassAdaptor &LHS, CGSCCToFunctionPassAdaptor &RHS) {
 543     using std::swap;
 544     swap(LHS.Pass, RHS.Pass);
 545   }
 546   CGSCCToFunctionPassAdaptor &operator=(CGSCCToFunctionPassAdaptor RHS) {
 547     swap(*this, RHS);
 548     return *this;
 549   }
 550
 551   /// \brief Runs the function pass across every function in the module.
 552   PreservedAnalyses run(LazyCallGraph::SCC &C, CGSCCAnalysisManager *AM) {
 553     FunctionAnalysisManager *FAM = nullptr;
 554     if (AM)
 555       // Setup the function analysis manager from its proxy.
 556       FAM = &AM->getResult<FunctionAnalysisManagerCGSCCProxy>(C).getManager();
 557
 558     PreservedAnalyses PA = PreservedAnalyses::all();
 559     for (LazyCallGraph::Node *N : C) {
 560       PreservedAnalyses PassPA = Pass.run(N->getFunction(), FAM);
 561
 562       // We know that the function pass couldn't have invalidated any other
 563       // function's analyses (that's the contract of a function pass), so
 564       // directly handle the function analysis manager's invalidation here.
 565       if (FAM)
 566         FAM->invalidate(N->getFunction(), PassPA);
 567
 568       // Then intersect the preserved set so that invalidation of module
 569       // analyses will eventually occur when the module pass completes.
 570       PA.intersect(std::move(PassPA));
 571     }
 572
 573     // By definition we preserve the proxy. This precludes *any* invalidation
 574     // of function analyses by the proxy, but that's OK because we've taken
 575     // care to invalidate analyses in the function analysis manager
 576     // incrementally above.
 577     // FIXME: We need to update the call graph here to account for any deleted
 578     // edges!
 579     PA.preserve<FunctionAnalysisManagerCGSCCProxy>();
 580     return PA;
 581   }
 582
 583   static StringRef name() { return "CGSCCToFunctionPassAdaptor"; }
 584
 585 private:
 586   FunctionPassT Pass;
 587 };
 588
 589 /// \brief A function to deduce a function pass type and wrap it in the
 590 /// templated adaptor.
 591 template <typename FunctionPassT>
 592 CGSCCToFunctionPassAdaptor<FunctionPassT>
 593 createCGSCCToFunctionPassAdaptor(FunctionPassT Pass) {
 594   return std::move(CGSCCToFunctionPassAdaptor<FunctionPassT>(std::move(Pass)));
 595 }
 596
 597 }
 598
 599 #endif