Keep tracking SIMD flag in assertion

Author: hez2010

src/coreclr/jit/assertionprop.cpp

@@ -1248,6 +1248,7 @@ AssertionIndex Compiler::optCreateAssertion(GenTree*         op1,
             assertion.op2.u1.iconVal = op2->AsIntCon()->gtIconVal;
             assertion.op2.vn         = optConservativeNormalVN(op2);
             assertion.op2.SetIconFlag(op2->GetIconHandleFlag());
+            assertion.op2.SetSimd(op2->IsCnsSimd());
 
             //
             // Ok everything has been set and the assertion looks good
@@ -1337,6 +1338,7 @@ AssertionIndex Compiler::optCreateAssertion(GenTree*         op1,
 
                         assertion.op2.u1.iconVal = iconVal;
                         assertion.op2.SetIconFlag(op2->GetIconHandleFlag(), op2->AsIntCon()->gtFieldSeq);
+                        assertion.op2.SetSimd(op2->IsCnsSimd());
                     }
                     else if (op2->gtOper == GT_CNS_LNG)
                     {
@@ -3340,6 +3342,10 @@ GenTree* Compiler::optConstantAssertionProp(AssertionDsc*        curAssertion,
             {
                 assert(varTypeIsIntegralOrI(tree));
                 newTree->BashToConst(curAssertion->op2.u1.iconVal, genActualType(tree));
+                if (curAssertion->op2.IsSimd())
+                {
+                    newTree->gtFlags |= GTF_ICON_SIMD_COUNT;
+                }
             }
             break;
 

src/coreclr/jit/compiler.h

@@ -7830,6 +7830,7 @@ class Compiler
             optOp2Kind kind; // a const or copy assertion
         private:
             uint16_t m_encodedIconFlags; // encoded icon gtFlags, don't use directly
+            bool     m_isSimd;
         public:
             ValueNum vn;
             struct IntVal
@@ -7871,6 +7872,14 @@ class Compiler
                 m_encodedIconFlags = flags >> iconMaskTzc;
                 u1.fieldSeq        = fieldSeq;
             }
+            bool IsSimd() const
+            {
+                return m_isSimd;
+            }
+            void SetSimd(bool simd)
+            {
+                m_isSimd = simd;
+            }
         } op2;
 
         bool IsCheckedBoundArithBound()

src/coreclr/jit/gentree.h

@@ -2236,6 +2236,11 @@ struct GenTree
         return (gtOper == GT_CNS_INT) ? (gtFlags & GTF_ICON_HDL_MASK) : GTF_EMPTY;
     }
 
+    bool IsCnsSimd() const
+    {
+        return (gtOper == GT_CNS_INT) ? ((gtFlags & GTF_ICON_SIMD_COUNT) != 0) : false;
+    }
+
     bool IsTlsIconHandle()
     {
         if (IsIconHandle())