TimurZaytsev

llvm/include/llvm/Transforms/TimurZaytsev/TimurZaytsev.h

@@ -0,0 +1,17 @@
+#ifndef LLVM_TRANSFORMS_TIMURZAYTSEV_H
+#define LLVM_TRANSFORMS_TIMURZAYTSEV_H
+
+#include "llvm/IR/PassManager.h"
+
+namespace llvm {
+
+class LPMUpdater;
+
+class TimurZaytsevPass : public PassInfoMixin<TimurZaytsevPass> {
+public:
+  PreservedAnalyses run(Function &F, FunctionAnalysisManager &AM); 
+};
+
+} // namespace llvm
+
+#endif // LLVM_TRANSFORMS_TIMURZAYTSEV_H 

llvm/lib/Passes/CMakeLists.txt

@@ -25,4 +25,5 @@ add_llvm_component_library(LLVMPasses
   TransformUtils
   Vectorize
   Instrumentation
+  TimurZaytsev
   )

llvm/lib/Passes/PassBuilder.cpp

@@ -23,6 +23,7 @@
 #include "llvm/Analysis/BlockFrequencyInfo.h"
 #include "llvm/Analysis/BranchProbabilityInfo.h"
 #include "llvm/Analysis/CFGPrinter.h"
+#include "llvm/Transforms/TimurZaytsev/TimurZaytsev.h"
 #include "llvm/Analysis/CFLAndersAliasAnalysis.h"
 #include "llvm/Analysis/CFLSteensAliasAnalysis.h"
 #include "llvm/Analysis/CGSCCPassManager.h"

llvm/lib/Passes/PassRegistry.def

@@ -253,6 +253,7 @@ FUNCTION_PASS("inject-tli-mappings", InjectTLIMappings())
 FUNCTION_PASS("instnamer", InstructionNamerPass())
 FUNCTION_PASS("loweratomic", LowerAtomicPass())
 FUNCTION_PASS("lower-expect", LowerExpectIntrinsicPass())
+FUNCTION_PASS("TimurZaytsev", TimurZaytsevPass())
 FUNCTION_PASS("lower-guard-intrinsic", LowerGuardIntrinsicPass())
 FUNCTION_PASS("lower-constant-intrinsics", LowerConstantIntrinsicsPass())
 FUNCTION_PASS("lower-widenable-condition", LowerWidenableConditionPass())

llvm/lib/Transforms/CMakeLists.txt

@@ -9,3 +9,4 @@ add_subdirectory(Hello)
 add_subdirectory(ObjCARC)
 add_subdirectory(Coroutines)
 add_subdirectory(CFGuard)
+add_subdirectory(TimurZaytsev)

llvm/lib/Transforms/TimurZaytsev/CMakeLists.txt

@@ -0,0 +1,10 @@
+add_llvm_component_library(LLVMTimurZaytsev
+  TimurZaytsev.cpp
+
+  DEPENDS
+  intrinsics_gen
+
+  LINK_COMPONENTS
+  Core
+  Support
+  ) 

llvm/lib/Transforms/TimurZaytsev/TimurZaytsev.cpp

@@ -0,0 +1,46 @@
+#include "llvm/Transforms/TimurZaytsev/TimurZaytsev.h"
+#include "llvm/IR/InstIterator.h"
+#include "llvm/Analysis/LoopInfo.h"
+
+#define DEBUG_TYPE "TimurZaytsev"
+#include "llvm/ADT/Statistic.h"
+
+using namespace llvm;
+
+ALWAYS_ENABLED_STATISTIC(Fun_Count, "Amount of functions");
+ALWAYS_ENABLED_STATISTIC(Op_Count, "Amount of arithmetic operations");
+ALWAYS_ENABLED_STATISTIC(B_Count, "Amount of blocks");
+ALWAYS_ENABLED_STATISTIC(Loop_Count, "Amount of loops");
+
+PreservedAnalyses TimurZaytsevPass::run(Function &F,
+                                      FunctionAnalysisManager &AM) {
+
+  Fun_Count++;
+  auto& LA = AM.getResult<LoopAnalysis>(F);
+  for (auto& L : LA) {
+    Loop_Count++;
+  }
+  for (const BasicBlock &BB : F) {
+    B_Count++;
+    for (const Instruction &I : BB) {
+      switch (I.getOpcode()) {
+      case Instruction::Add:
+      case Instruction::Sub:
+      case Instruction::Mul:
+      case Instruction::FAdd:
+      case Instruction::FSub:
+      case Instruction::FMul:
+      case Instruction::FDiv:
+      case Instruction::SDiv:
+      case Instruction::UDiv:
+        Op_Count++;
+        break;
+      }
+    }
+  }
+  errs() << " Amount of arithmetic operations " << Op_Count << "\n";
+  errs() << " Amount of loops " << Loop_Count << "\n";
+  errs() << " Amount of functions " << Fun_Count << "\n";
+  errs() << " Amount of blocks " << B_Count << "\n";
+  return PreservedAnalyses::all();
+}