[llvm] [SCEV] Use power of two facts involving vscale when inferring wrap flags (PR #101380)

[Nikita Popov via llvm-commits]

================
@@ -10845,6 +10843,25 @@ bool ScalarEvolution::isKnownNonZero(const SCEV *S) {
   return getUnsignedRangeMin(S) != 0;
 }
 
+bool ScalarEvolution::isKnownToBeAPowerOfTwo(const SCEV *S, bool OrZero) {
+  auto nonRecursive = [this](const SCEV *S) {
+    if (auto *C = dyn_cast<SCEVConstant>(S))
+      return C->getAPInt().isPowerOf2();
+    // The vscale_range indicates vscale is a power-of-two.
+    return S->getSCEVType() == scVScale && F.hasFnAttribute(Attribute::VScaleRange);;
----------------
nikic wrote:

```suggestion
    return isa<SCEVVScale>(S) && F.hasFnAttribute(Attribute::VScaleRange);
```

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