1 //===- InlineSizeEstimatorAnalysis.h - ML size estimator --------*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 
10 #ifndef LLVM_ANALYSIS_INLINESIZEESTIMATORANALYSIS_H
11 #define LLVM_ANALYSIS_INLINESIZEESTIMATORANALYSIS_H
12 
13 #include "llvm/IR/PassManager.h"
14 
15 namespace llvm {
16 class Function;
17 
18 class TFModelEvaluator;
19 class InlineSizeEstimatorAnalysis
20     : public AnalysisInfoMixin<InlineSizeEstimatorAnalysis> {
21 public:
22   InlineSizeEstimatorAnalysis();
23   InlineSizeEstimatorAnalysis(InlineSizeEstimatorAnalysis &&);
24   ~InlineSizeEstimatorAnalysis();
25 
26   static AnalysisKey Key;
27   using Result = std::optional<size_t>;
28   Result run(const Function &F, FunctionAnalysisManager &FAM);
29   static bool isEvaluatorRequested();
30 
31 private:
32   std::unique_ptr<TFModelEvaluator> Evaluator;
33 };
34 
35 class InlineSizeEstimatorAnalysisPrinterPass
36     : public PassInfoMixin<InlineSizeEstimatorAnalysisPrinterPass> {
37   raw_ostream &OS;
38 
39 public:
40   explicit InlineSizeEstimatorAnalysisPrinterPass(raw_ostream &OS) : OS(OS) {}
41 
42   PreservedAnalyses run(Function &F, FunctionAnalysisManager &AM);
43 
44   static bool isRequired() { return true; }
45 };
46 } // namespace llvm
47 #endif // LLVM_ANALYSIS_INLINESIZEESTIMATORANALYSIS_H
48