11d0244aeSSameer Sahasrabuddhe //===- CycleAnalysis.cpp - Compute CycleInfo for LLVM IR ------------------===//
21d0244aeSSameer Sahasrabuddhe //
31d0244aeSSameer Sahasrabuddhe // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
41d0244aeSSameer Sahasrabuddhe // See https://llvm.org/LICENSE.txt for license information.
51d0244aeSSameer Sahasrabuddhe // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
61d0244aeSSameer Sahasrabuddhe //
71d0244aeSSameer Sahasrabuddhe //===----------------------------------------------------------------------===//
81d0244aeSSameer Sahasrabuddhe 
91d0244aeSSameer Sahasrabuddhe #include "llvm/Analysis/CycleAnalysis.h"
101d0244aeSSameer Sahasrabuddhe #include "llvm/ADT/GenericCycleImpl.h"
11*71c3a551Sserge-sans-paille #include "llvm/IR/CFG.h" // for successors found by ADL in GenericCycleImpl.h
121d0244aeSSameer Sahasrabuddhe #include "llvm/InitializePasses.h"
131d0244aeSSameer Sahasrabuddhe 
141d0244aeSSameer Sahasrabuddhe using namespace llvm;
151d0244aeSSameer Sahasrabuddhe 
16*71c3a551Sserge-sans-paille namespace llvm {
17*71c3a551Sserge-sans-paille class Module;
18*71c3a551Sserge-sans-paille }
19*71c3a551Sserge-sans-paille 
201d0244aeSSameer Sahasrabuddhe template class llvm::GenericCycleInfo<SSAContext>;
211d0244aeSSameer Sahasrabuddhe template class llvm::GenericCycle<SSAContext>;
221d0244aeSSameer Sahasrabuddhe 
run(Function & F,FunctionAnalysisManager &)231d0244aeSSameer Sahasrabuddhe CycleInfo CycleAnalysis::run(Function &F, FunctionAnalysisManager &) {
241d0244aeSSameer Sahasrabuddhe   CycleInfo CI;
251d0244aeSSameer Sahasrabuddhe   CI.compute(F);
261d0244aeSSameer Sahasrabuddhe   return CI;
271d0244aeSSameer Sahasrabuddhe }
281d0244aeSSameer Sahasrabuddhe 
291d0244aeSSameer Sahasrabuddhe AnalysisKey CycleAnalysis::Key;
301d0244aeSSameer Sahasrabuddhe 
CycleInfoPrinterPass(raw_ostream & OS)311d0244aeSSameer Sahasrabuddhe CycleInfoPrinterPass::CycleInfoPrinterPass(raw_ostream &OS) : OS(OS) {}
321d0244aeSSameer Sahasrabuddhe 
run(Function & F,FunctionAnalysisManager & AM)331d0244aeSSameer Sahasrabuddhe PreservedAnalyses CycleInfoPrinterPass::run(Function &F,
341d0244aeSSameer Sahasrabuddhe                                             FunctionAnalysisManager &AM) {
351d0244aeSSameer Sahasrabuddhe   OS << "CycleInfo for function: " << F.getName() << "\n";
361d0244aeSSameer Sahasrabuddhe   AM.getResult<CycleAnalysis>(F).print(OS);
371d0244aeSSameer Sahasrabuddhe 
381d0244aeSSameer Sahasrabuddhe   return PreservedAnalyses::all();
391d0244aeSSameer Sahasrabuddhe }
401d0244aeSSameer Sahasrabuddhe 
411d0244aeSSameer Sahasrabuddhe //===----------------------------------------------------------------------===//
421d0244aeSSameer Sahasrabuddhe //  CycleInfoWrapperPass Implementation
431d0244aeSSameer Sahasrabuddhe //===----------------------------------------------------------------------===//
441d0244aeSSameer Sahasrabuddhe //
451d0244aeSSameer Sahasrabuddhe // The implementation details of the wrapper pass that holds a CycleInfo
461d0244aeSSameer Sahasrabuddhe // suitable for use with the legacy pass manager.
471d0244aeSSameer Sahasrabuddhe //
481d0244aeSSameer Sahasrabuddhe //===----------------------------------------------------------------------===//
491d0244aeSSameer Sahasrabuddhe 
501d0244aeSSameer Sahasrabuddhe char CycleInfoWrapperPass::ID = 0;
511d0244aeSSameer Sahasrabuddhe 
CycleInfoWrapperPass()521d0244aeSSameer Sahasrabuddhe CycleInfoWrapperPass::CycleInfoWrapperPass() : FunctionPass(ID) {
531d0244aeSSameer Sahasrabuddhe   initializeCycleInfoWrapperPassPass(*PassRegistry::getPassRegistry());
541d0244aeSSameer Sahasrabuddhe }
551d0244aeSSameer Sahasrabuddhe 
561d0244aeSSameer Sahasrabuddhe INITIALIZE_PASS_BEGIN(CycleInfoWrapperPass, "cycles", "Cycle Info Analysis",
571d0244aeSSameer Sahasrabuddhe                       true, true)
581d0244aeSSameer Sahasrabuddhe INITIALIZE_PASS_END(CycleInfoWrapperPass, "cycles", "Cycle Info Analysis", true,
591d0244aeSSameer Sahasrabuddhe                     true)
601d0244aeSSameer Sahasrabuddhe 
getAnalysisUsage(AnalysisUsage & AU) const611d0244aeSSameer Sahasrabuddhe void CycleInfoWrapperPass::getAnalysisUsage(AnalysisUsage &AU) const {
621d0244aeSSameer Sahasrabuddhe   AU.setPreservesAll();
631d0244aeSSameer Sahasrabuddhe }
641d0244aeSSameer Sahasrabuddhe 
runOnFunction(Function & Func)651d0244aeSSameer Sahasrabuddhe bool CycleInfoWrapperPass::runOnFunction(Function &Func) {
661d0244aeSSameer Sahasrabuddhe   CI.clear();
671d0244aeSSameer Sahasrabuddhe 
681d0244aeSSameer Sahasrabuddhe   F = &Func;
691d0244aeSSameer Sahasrabuddhe   CI.compute(Func);
701d0244aeSSameer Sahasrabuddhe   return false;
711d0244aeSSameer Sahasrabuddhe }
721d0244aeSSameer Sahasrabuddhe 
print(raw_ostream & OS,const Module *) const731d0244aeSSameer Sahasrabuddhe void CycleInfoWrapperPass::print(raw_ostream &OS, const Module *) const {
741d0244aeSSameer Sahasrabuddhe   OS << "CycleInfo for function: " << F->getName() << "\n";
751d0244aeSSameer Sahasrabuddhe   CI.print(OS);
761d0244aeSSameer Sahasrabuddhe }
771d0244aeSSameer Sahasrabuddhe 
releaseMemory()781d0244aeSSameer Sahasrabuddhe void CycleInfoWrapperPass::releaseMemory() {
791d0244aeSSameer Sahasrabuddhe   CI.clear();
801d0244aeSSameer Sahasrabuddhe   F = nullptr;
811d0244aeSSameer Sahasrabuddhe }
82