[llvm] Fix/aarch64 memset dup optimization (PR #166030)

[Osama Abdelkader via llvm-commits]

================
@@ -18341,10 +18342,12 @@ EVT AArch64TargetLowering::getOptimalMemOpType(
            Fast;
   };
 
-  if (CanUseNEON && Op.isMemset() && !IsSmallMemset &&
-      AlignmentIsAcceptable(MVT::v16i8, Align(16)))
+  // For non-zero memset, use NEON even for smaller sizes as dup + scalar store
+  // is efficient
+  if (CanUseNEON && Op.isMemset() && !IsSmallZeroMemset)
----------------
osamakader wrote:

Done.

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