[inline] Heuristic

lyh552506 · Aug 20, 2024 · 2be7836 · 2be7836
1 parent 4a740ee
commit 2be7836
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/include/lib/Singleton.hpp b/include/lib/Singleton.hpp
@@ -12,4 +12,8 @@ struct IR_CONSTDECL_FLAG
 
 struct Enable_Parallel{
     bool flag=false;
+};
+
+struct Inline_Recursion{
+    bool flag=false;
 };
diff --git a/ir/opt/Inline.cpp b/ir/opt/Inline.cpp
@@ -129,12 +129,12 @@ bool NoRecursive::CanBeInlined(CallInst *call)
     auto &&master = call->GetParent()->GetParent();
     if (slave->tag == Function::Tag::ParallelBody || master->tag == Function::Tag::UnrollBody || slave->tag == Function::Tag::BuildIn)
         return false;
-    if (InlineRecursion)
+    if (Singleton<Inline_Recursion>().flag)
     {
         static int InlineTimes = 0;
         if (!master->isRecursive() && !slave->isRecursive())
             return true;
-        if (InlineTimes < 5)
+        if (InlineTimes<3&&(master->Size()+slave->Size())*3<100)
         {
             InlineTimes++;
             return true;

diff --git a/ir/opt/new_passManager.cpp b/ir/opt/new_passManager.cpp
@@ -193,6 +193,7 @@ void _PassManager::RunOnLevel() {
         PassChangedBegin(curfunc) 
       }
 
+      Singleton<Inline_Recursion>().flag=true;
       while (RunImpl<Inliner>(module, AM)) {
         RunLevelPass(cfgSimplify, curfunc, modified)
             RunImpl<DeadArgsElimination>(module, AM);