[loop count assumptions] by default enable loop iteration assumption …

…for AIE targets

clang/test/CodeGen/aie/peel-itercount.c

@@ -14,10 +14,8 @@
 // and that the itercounts have been updated appropriately
 
 // CHECK-LABEL: loop28_37
-// CHECK: entry
-// CHECK: for.body.preheader:
 // CHECK: for.body.peel.next:
-// CHECK: for.cond.cleanup: 
+// CHECK: for.cond.cleanup:
 // CHECK: for.body:
 // CHECK: !llvm.loop !6
 // CHECK: !6 = distinct !{!6, !7, !8, !9}

llvm/lib/Target/AIE/AIEBaseTargetMachine.cpp

@@ -43,6 +43,7 @@
 #include "llvm/Transforms/IPO/GlobalDCE.h"
 #include "llvm/Transforms/IPO/Internalize.h"
 #include "llvm/Transforms/Scalar.h"
+#include "llvm/Transforms/Utils/LoopIterCountAssumptions.h"
 
 using namespace llvm;
 
@@ -56,6 +57,11 @@ static cl::opt<bool>
                          cl::desc("Enable tail merging for AIE."),
                          cl::init(false), cl::Hidden);
 
+static cl::opt<bool> EnableLoopIterCountToAssumptions(
+    "enable-loop-iter-count-assumptions", cl::Hidden, cl::init(true),
+    cl::desc(
+        "Enable Conversion of Loop Iteration Count Metadata to Assumptions."));
+
 // Option to run internalize pass.
 static cl::opt<bool> InternalizeSymbols(
     "aie-internalize-symbols",
@@ -179,6 +185,15 @@ void AIEBaseTargetMachine::registerPassBuilderCallbacks(
           }
         });
   }
+
+  if (EnableLoopIterCountToAssumptions) {
+    PB.registerPipelineEarlySimplificationEPCallback(
+        [](ModulePassManager &PM, OptimizationLevel) {
+          PM.addPass(createModuleToFunctionPassAdaptor(
+              createFunctionToLoopPassAdaptor(LoopIterCountAssumptions(),
+                                              /*UseMemorySSA*/ true)));
+        });
+  }
 }
 
 AIEBasePassConfig::AIEBasePassConfig(LLVMTargetMachine &TM, PassManagerBase &PM)