JIT: Determine which ARM64 HWIntrinsics will throw ArgumentOufOfRange…

…Exceptions (#105527)

src/coreclr/jit/gentree.cpp

@@ -7201,6 +7201,8 @@ ExceptionSetFlags GenTree::OperExceptions(Compiler* comp)
 #ifdef FEATURE_HW_INTRINSICS
         case GT_HWINTRINSIC:
         {
+            assert((gtFlags & GTF_HW_USER_CALL) == 0);
+
             GenTreeHWIntrinsic* hwIntrinsicNode = this->AsHWIntrinsic();
 
             if (hwIntrinsicNode->OperIsMemoryLoadOrStore())
@@ -7239,6 +7241,15 @@ bool GenTree::OperMayThrow(Compiler* comp)
         helper = comp->eeGetHelperNum(this->AsCall()->gtCallMethHnd);
         return ((helper == CORINFO_HELP_UNDEF) || !comp->s_helperCallProperties.NoThrow(helper));
     }
+#ifdef FEATURE_HW_INTRINSICS
+    else if (OperIsHWIntrinsic())
+    {
+        if ((gtFlags & GTF_HW_USER_CALL) != 0)
+        {
+            return true;
+        }
+    }
+#endif // FEATURE_HW_INTRINSICS
 
     return OperExceptions(comp) != ExceptionSetFlags::None;
 }
@@ -20118,7 +20129,7 @@ void GenTreeJitIntrinsic::SetMethodHandle(Compiler*                          com
                                           CORINFO_METHOD_HANDLE methodHandle R2RARG(CORINFO_CONST_LOOKUP entryPoint))
 {
     assert(OperIsHWIntrinsic() && !IsUserCall());
-    gtFlags |= GTF_HW_USER_CALL;
+    gtFlags |= (GTF_HW_USER_CALL | GTF_EXCEPT | GTF_CALL);
 
     size_t operandCount = GetOperandCount();
 

src/coreclr/jit/rationalize.cpp

@@ -505,7 +505,7 @@ void Rationalizer::RewriteHWIntrinsicAsUserCall(GenTree** use, ArrayStack<GenTre
                                                    immUpperBound, hasFullRangeImm, &useFallback))
                 {
                     // We're already in the right shape, so just stop tracking ourselves as a user call
-                    hwintrinsic->gtFlags &= ~GTF_HW_USER_CALL;
+                    hwintrinsic->gtFlags &= ~(GTF_HW_USER_CALL | GTF_EXCEPT | GTF_CALL);
                     return;
                 }
             }

src/tests/JIT/Regression/JitBlue/Runtime_105474/Runtime_105474.cs

@@ -0,0 +1,44 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+
+using System;
+using System.Runtime.CompilerServices;
+using System.Runtime.Intrinsics;
+using System.Runtime.Intrinsics.Arm;
+using System.Runtime.Intrinsics.X86;
+using Xunit;
+
+#nullable disable
+
+public class Runtime_105474_A
+{
+    private void Method0()
+    {
+        Vector128<ulong> vr0 = Vector128.CreateScalar(1698800584428641629UL);
+        AdvSimd.ShiftLeftLogicalSaturate(vr0, 229);
+    }
+
+    private void Method1()
+    {
+        Vector128<float> vr1 = default;
+        Avx.Compare(vr1, vr1, (FloatComparisonMode)255);
+    }
+
+    [Fact]
+    public static void TestEntryPointArm()
+    {
+        if (AdvSimd.IsSupported)
+        {
+            Assert.Throws<ArgumentOutOfRangeException>(() => new Runtime_105474_A().Method0());
+        }
+    }
+
+    [Fact]
+    public static void TestEntryPoint()
+    {
+        if (Avx.IsSupported)
+        {
+            Assert.Throws<ArgumentOutOfRangeException>(() => new Runtime_105474_A().Method1());
+        }
+    }
+}

src/tests/JIT/Regression/JitBlue/Runtime_105474/Runtime_105474.csproj

@@ -0,0 +1,8 @@
+<Project Sdk="Microsoft.NET.Sdk">
+  <PropertyGroup>
+    <Optimize>True</Optimize>
+  </PropertyGroup>
+  <ItemGroup>
+    <Compile Include="$(MSBuildProjectName).cs" />
+  </ItemGroup>
+</Project>