[llvm] [AArch64] Add flag to control unrolling for small multi-exit loops (PR #131998)

[Igor Kirillov via llvm-commits]

================
@@ -4237,6 +4242,81 @@ getFalkorUnrollingPreferences(Loop *L, ScalarEvolution &SE,
   }
 }
 
+static bool shouldUnrollLoopWithInstruction(Instruction &I,
+                                            AArch64TTIImpl &TTI) {
+  // Don't unroll vectorised loop.
+  if (I.getType()->isVectorTy())
+    return false;
+
+  if (isa<CallInst>(I) || isa<InvokeInst>(I)) {
----------------
igogo-x86 wrote:

Since we are only working with `CallBase` below, we can rewrite it as:
```
if (auto *CB = dyn_cast<CallBase>(&I)) {
  if (const Function *F = CB->getCalledFunction())
     ....
}
```

https://github.com/llvm/llvm-project/pull/131998