Misaki
155d7b0fbd
- ISPMDLane: add MaskGather, MaskStore, Scatter, MaskScatter; update MaskLoad/Gather signatures for hardware parity - WideLane/ScalarLane: implement new methods with HW/fallback logic - MathV: gather/mask-gather now delegate to lane methods - Vector2/3/4: add CompressStore, Scatter, MaskScatter - SPMD jobs/tests/README: migrate to new APIs for correctness - Use Unsafe.BitCast instead of Unsafe.As/AsRef - Add SPMDUtility for gather index extraction - Job system: add ICustomJob<TSelf>, ScheduleCustom overload - FreeList concurrency obsolete; always thread-safe - NuGet: include LICENSE/README, set license/readme in .csproj - Docs: update SPMD usage, clarify safety notes - Minor: doc fixes, CompressStore test improvements
64 lines
2.8 KiB
C#
64 lines
2.8 KiB
C#
using System.Numerics;
|
|
using System.Runtime.CompilerServices;
|
|
|
|
namespace Misaki.HighPerformance.Mathematics.SPMD;
|
|
|
|
public readonly unsafe partial struct WideLane<TNumber> : ISPMDLane<WideLane<TNumber>, TNumber>
|
|
where TNumber : unmanaged, INumber<TNumber>, IBinaryNumber<TNumber>, IMinMaxValue<TNumber>, IBitwiseOperators<TNumber, TNumber, TNumber>
|
|
{
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public TOther Cast<TOther, TOtherNumber>()
|
|
where TOther : ISPMDLane<TOther, TOtherNumber>
|
|
where TOtherNumber : unmanaged, INumber<TOtherNumber>, IBinaryNumber<TOtherNumber>, IMinMaxValue<TOtherNumber>, IBitwiseOperators<TOtherNumber, TOtherNumber, TOtherNumber>
|
|
{
|
|
if (typeof(TNumber) == typeof(float) && typeof(TOtherNumber) == typeof(int))
|
|
{
|
|
return Unsafe.BitCast<Vector<int>, TOther>(Vector.ConvertToInt32(Unsafe.BitCast<Vector<TNumber>, Vector<float>>(value)));
|
|
}
|
|
|
|
if (typeof(TNumber) == typeof(float) && typeof(TOtherNumber) == typeof(uint))
|
|
{
|
|
return Unsafe.BitCast<Vector<uint>, TOther>(Vector.ConvertToUInt32(Unsafe.BitCast<Vector<TNumber>, Vector<float>>(value)));
|
|
}
|
|
|
|
if (typeof(TNumber) == typeof(double) && typeof(TOtherNumber) == typeof(long))
|
|
{
|
|
return Unsafe.BitCast<Vector<long>, TOther>(Vector.ConvertToInt64(Unsafe.BitCast<Vector<TNumber>, Vector<double>>(value)));
|
|
}
|
|
|
|
if (typeof(TNumber) == typeof(double) && typeof(TOtherNumber) == typeof(ulong))
|
|
{
|
|
return Unsafe.BitCast<Vector<ulong>, TOther>(Vector.ConvertToUInt64(Unsafe.BitCast<Vector<TNumber>, Vector<double>>(value)));
|
|
}
|
|
|
|
if (typeof(TNumber) == typeof(int) && typeof(TOtherNumber) == typeof(float))
|
|
{
|
|
return Unsafe.BitCast<Vector<float>, TOther>(Vector.ConvertToSingle(Unsafe.BitCast<Vector<TNumber>, Vector<int>>(value)));
|
|
}
|
|
|
|
if (typeof(TNumber) == typeof(uint) && typeof(TOtherNumber) == typeof(float))
|
|
{
|
|
return Unsafe.BitCast<Vector<float>, TOther>(Vector.ConvertToSingle(Unsafe.BitCast<Vector<TNumber>, Vector<uint>>(value)));
|
|
}
|
|
|
|
if (typeof(TNumber) == typeof(long) && typeof(TOtherNumber) == typeof(double))
|
|
{
|
|
return Unsafe.BitCast<Vector<double>, TOther>(Vector.ConvertToDouble(Unsafe.BitCast<Vector<TNumber>, Vector<long>>(value)));
|
|
}
|
|
|
|
if (typeof(TNumber) == typeof(ulong) && typeof(TOtherNumber) == typeof(double))
|
|
{
|
|
return Unsafe.BitCast<Vector<double>, TOther>(Vector.ConvertToDouble(Unsafe.BitCast<Vector<TNumber>, Vector<ulong>>(value)));
|
|
}
|
|
|
|
var casted = stackalloc TOtherNumber[LaneWidth];
|
|
for (var i = 0; (i < LaneWidth) && (i < TOther.LaneWidth); i++)
|
|
{
|
|
casted[i] = TOtherNumber.CreateTruncating(value[i]);
|
|
}
|
|
|
|
return TOther.Load(casted);
|
|
}
|
|
}
|
|
|