[SLP] Skip MMRA metadata if instruction can't have it (#173319)

Exposed by recent commit in e644f06, but the wrong propagation
might be applied in other situations too.

Without this, we will get verification errors with e644f06.

!mmra metadata attached to unexpected instruction kind
%3 = fmul <2 x float> %2, <float 1.000000e+00, float 0.000000e+00>,
!mmra !1

Author: jsji

llvm/lib/Analysis/VectorUtils.cpp

@@ -1084,6 +1084,10 @@ Instruction *llvm::propagateMetadata(Instruction *Inst, ArrayRef<Value *> VL) {
   getMetadataToPropagate(cast<Instruction>(VL[0]), Metadata);
 
   for (auto &[Kind, MD] : Metadata) {
+    // Skip MMRA metadata if the instruction cannot have it.
+    if (Kind == LLVMContext::MD_mmra && !canInstructionHaveMMRAs(*Inst))
+      continue;
+
     for (int J = 1, E = VL.size(); MD && J != E; ++J) {
       const Instruction *IJ = cast<Instruction>(VL[J]);
       MDNode *IMD = IJ->getMetadata(Kind);

llvm/test/Transforms/SLPVectorizer/X86/propagate-mmra.ll

@@ -0,0 +1,32 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 6
+; RUN: opt -passes=slp-vectorizer -S -mtriple=x86_64-unknown -slp-vectorize-hor < %s | FileCheck %s
+
+define void @foo() {
+; CHECK-LABEL: define void @foo() {
+; CHECK-NEXT:  [[_PREHEADER16_PREHEADER:.*:]]
+; CHECK-NEXT:    [[TMP0:%.*]] = load float, ptr null, align 4, !mmra [[META0:![0-9]+]]
+; CHECK-NEXT:    [[TMP2:%.*]] = fmul float [[TMP0]], 0.000000e+00
+; CHECK-NEXT:    [[TMP1:%.*]] = insertelement <2 x float> poison, float [[TMP0]], i32 0
+; CHECK-NEXT:    [[TMP3:%.*]] = insertelement <2 x float> [[TMP1]], float [[TMP2]], i32 1
+; CHECK-NEXT:    [[TMP4:%.*]] = fadd <2 x float> [[TMP3]], zeroinitializer
+; CHECK-NEXT:    [[TMP5:%.*]] = select <2 x i1> zeroinitializer, <2 x float> [[TMP4]], <2 x float> zeroinitializer
+; CHECK-NEXT:    store <2 x float> [[TMP5]], ptr null, align 16
+; CHECK-NEXT:    ret void
+;
+.preheader16.preheader:
+  %0 = load float, ptr null, align 4, !mmra !0
+  %1 = fadd float %0, 0.000000e+00
+  %2 = select i1 false, float %1, float 0.000000e+00
+  store float %2, ptr null, align 16
+  %3 = fmul float %0, 0.000000e+00
+  %4 = fadd float %3, 0.000000e+00
+  %5 = select i1 false, float %4, float 0.000000e+00
+  %.sroa.34.96..sroa_idx = getelementptr i8, ptr null, i64 4
+  store float %5, ptr %.sroa.34.96..sroa_idx, align 4
+  ret void
+}
+
+!0 = !{!"scope", !"device"}
+;.
+; CHECK: [[META0]] = !{!"scope", !"device"}
+;.