Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 10 additions & 0 deletions src/coreclr/jit/gentree.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -20817,6 +20817,16 @@ GenTree* Compiler::gtNewSimdBinOpNode(
if (op2->IsCnsIntOrI())
{
op2->AsIntCon()->gtIconVal &= shiftCountMask;
#ifdef TARGET_ARM64
// On ARM64, ShiftRight* intrinsics cannot encode a shift value of zero,
// so use the generic Shift* fallback intrinsic.
// GenTreeHWIntrinsic::GetHWIntrinsicIdForBinOp will see that the immediate node is not const,
// and return the correct fallback intrinsic.
if ((op != GT_LSH) && (op2->AsIntCon()->IconValue() == 0))
{
op2 = gtNewZeroConNode(type);
Copy link
Copy Markdown
Member

@tannergooding tannergooding Sep 6, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nit, we prefer the direct zero constant helper in cases like this:

Suggested change
op2 = gtNewSimdCreateBroadcastNode(type, op2, simdBaseJitType, simdSize);
op2 = gtNewZeroConNode(type);

}
#endif // TARGET_ARM64
}
else
{
Expand Down
14 changes: 2 additions & 12 deletions src/coreclr/jit/hwintrinsiccodegenarm64.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -434,18 +434,8 @@ void CodeGen::genHWIntrinsic(GenTreeHWIntrinsic* node)
for (helper.EmitBegin(); !helper.Done(); helper.EmitCaseEnd())
{
const int shiftAmount = helper.ImmValue();

if (shiftAmount == 0)
{
// TODO: Use emitIns_Mov instead.
// We do not use it currently because it will still elide the 'mov'
// even if 'canSkip' is false. We cannot elide the 'mov' here.
GetEmitter()->emitIns_R_R_R(INS_mov, emitTypeSize(node), targetReg, reg, reg);
}
else
{
GetEmitter()->emitIns_R_R_I(ins, emitSize, targetReg, reg, shiftAmount, opt);
}
assert((shiftAmount != 0) || (intrin.category == HW_Category_ShiftLeftByImmediate));
GetEmitter()->emitIns_R_R_I(ins, emitSize, targetReg, reg, shiftAmount, opt);
}
};

Expand Down
45 changes: 45 additions & 0 deletions src/tests/JIT/Regression/JitBlue/Runtime_107173/Runtime_107173.cs
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

// Generated by Fuzzlyn v2.4 on 2024-08-26 23:38:13
// Run on Arm64 Linux
// Seed: 8716802894387291290-vectort,vector64,vector128,armadvsimd,armadvsimdarm64,armaes,armarmbase,armarmbasearm64,armcrc32,armcrc32arm64,armdp,armrdm,armrdmarm64,armsha1,armsha256
// Reduced from 19.5 KiB to 0.5 KiB in 00:00:27
// Debug: Outputs <0, 0, 0, 0>
// Release: Outputs <0, 0, 4457472, 0>
using System;
using System.Numerics;
using System.Runtime.Intrinsics;
using System.Runtime.Intrinsics.Arm;
using Xunit;

public class C0
{
public ushort F2;
public ushort F8;
}

public class Runtime_107173
{
public static C0 s_8 = new C0();

[Fact]
public static void TestLeftShift()
{
if (AdvSimd.IsSupported)
{
var vr6 = s_8.F8;
var vr7 = s_8.F2;
var vr8 = Vector64.Create(vr6, vr7, 0, 0);
Vector128<uint> vr9 = AdvSimd.ShiftLeftLogicalWideningLower(vr8, 0);
Assert.Equal(vr9, Vector128<uint>.Zero);
}
}

[Fact]
public static void TestRightShift()
{
var result = Vector128<byte>.AllBitsSet >> 8;
Assert.Equal(result, Vector128<byte>.AllBitsSet);
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
<Project Sdk="Microsoft.NET.Sdk">
<PropertyGroup>
<Optimize>True</Optimize>
</PropertyGroup>
<ItemGroup>
<Compile Include="$(MSBuildProjectName).cs" />
</ItemGroup>
</Project>