merge from giga

2025-09-19 00:30:02 +08:00
parent 36f8fb1107 8007eee846
commit aded199bc6
40 changed files with 732 additions and 272 deletions
@@ -0,0 +1,148 @@
+//-----------------------------------------------------------------------
+// -----------------------------------------------------------------------
+// file="MatrixParallel.TaskWrapFactory.cs"
+// Project: StgSharp
+// AuthorGroup: Nitload Space
+// Copyright (c) Nitload Space. All rights reserved.
+//     
+// Permission is hereby granted, free of charge, to any person 
+// obtaining a copy of this software and associated documentation 
+// files (the “Software”), to deal in the Software without restriction, 
+// including without limitation the rights to use, copy, modify, merge,
+// publish, distribute, sublicense, and/or sell copies of the Software, 
+// and to permit persons to whom the Software is furnished to do so, 
+// subject to the following conditions:
+//     
+// The above copyright notice and 
+// this permission notice shall be included in all copies 
+// or substantial portions of the Software.
+//     
+// THE SOFTWARE IS PROVIDED “AS IS”, 
+// WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, 
+// INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. 
+// IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, 
+// DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, 
+// ARISING FROM, OUT OF OR IN CONNECTION WITH 
+// THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+//     
+// -----------------------------------------------------------------------
+// -----------------------------------------------------------------------
+using StgSharp.HighPerformance;
+using StgSharp.HighPerformance.Memory;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Runtime.CompilerServices;
+using System.Text;
+using System.Threading;
+using System.Threading.Tasks;
+
+using hlsfAllocator = StgSharp.HighPerformance.Memory.HybridLayerSegregatedFitAllocator;
+
+namespace StgSharp.Mathematics
+{
+    internal static unsafe class MatrixParallelFactory
+    {
+
+        private static hlsfAllocator _intermediateResult;
+
+        private static
+#if NET9_0_OR_GREATER
+            Lock
+#else
+            object
+#endif
+            _hlsfLock = new();
+
+        private static SlabAllocator<ScalarPacket> _scalarAllocator ;
+
+        private static SlabAllocator<MatrixParallelTaskPackage> _wrapAllocator ;
+
+        [MethodImpl(MethodImplOptions.AggressiveInlining)]
+        public static MatrixParallelTaskPackage* CreateBaseTask()
+        {
+            MatrixParallelTaskPackage* p = (MatrixParallelTaskPackage*)_wrapAllocator.Allocate();
+            Span<byte> s = new Span<byte>((byte*)p, sizeof(MatrixParallelTaskPackage));
+            s.Fill(0);
+            return p;
+        }
+
+        [MethodImpl(MethodImplOptions.AggressiveInlining)]
+        public static ScalarPacket* CreateScalarPacket()
+        {
+            ScalarPacket* p = (ScalarPacket*)_scalarAllocator.Allocate();
+            return p;
+        }
+
+<<<<<<< HEAD:StgSharp.Common/Mathematics/Linear/MatrixParallel.TaskWrapFactory.cs
+        public static MatrixParallelTaskPackage* FromTaskWrap(MatrixParallelTaskPackage* source, int offset)
+        {
+            MatrixParallelTaskPackage* p = (MatrixParallelTaskPackage*)_wrapAllocator.Allocate();
+            /*
+            *p = *source;
+            p->Left += offset * p->LeftStride;
+            p->Right += offset * p->RightStride;
+            p->Result += offset * p->ResultStride;
+            /**/
+            return p;
+        }
+
+=======
+        [MethodImpl(MethodImplOptions.AggressiveInlining)]
+>>>>>>> 8007eee8461498a85a9515e8fd076f03cb552b64:StgSharp.Common/Mathematics/Linear/MatrixParallel.TaskFactory.cs
+        public static void Release(ScalarPacket* source)
+        {
+            _scalarAllocator.Free((nuint)source);
+        }
+
+        [MethodImpl(MethodImplOptions.AggressiveInlining)]
+        public static void Release(MatrixParallelTaskPackage* source)
+        {
+            if (source->Scalar != null) {
+                _scalarAllocator.Free((nuint)source->Scalar);
+            }
+            _wrapAllocator.Free((nuint)source);
+        }
+
+        [MethodImpl(MethodImplOptions.AggressiveInlining)]
+        internal static void Init()
+        {
+<<<<<<< HEAD:StgSharp.Common/Mathematics/Linear/MatrixParallel.TaskWrapFactory.cs
+            int count = group.Length;
+            group[0] = (nint)source;
+            int l = source->LeftPrimStride, r = source->RightPrimStride, res = source->ResultPrimStride, c = source->PrimCount / count;
+            source->LeftPrimStride *= count;
+            source->RightPrimStride *= count;
+            source->ResultPrimStride *= count;
+            source->PrimCount = c;
+            int remain = source->PrimCount - (c * count);
+            for (int i = 1; i < count; i++)
+            {
+                nint handle = (nint)_wrapAllocator.Allocate();
+                MatrixParallelTaskPackage* p = (MatrixParallelTaskPackage*)handle;
+                *p = *source;
+                p->Left += l * i;
+                p->Right += r * i;
+                p->Result += res * i;
+                p->PrimCount = c;
+                group[i] = handle;
+            }
+
+            // Distribute remaining work to the last few tasks
+            for (int i = 0; i < remain; i++)
+            {
+                int targetIndex = count - 1 - i;
+                if (targetIndex >= 0) {
+                    ((MatrixParallelTaskPackage*)group[targetIndex])->PrimCount++;
+                }
+            }
+=======
+            _scalarAllocator = SlabAllocator<ScalarPacket>.Create(64, SlabBufferLayout.Chunked, true);
+            _wrapAllocator = SlabAllocator<MatrixParallelTaskPackage>.Create(64, SlabBufferLayout.Chunked, true);
+            _intermediateResult = new hlsfAllocator((nuint)(sizeof(MatrixKernel) * 64));
+>>>>>>> 8007eee8461498a85a9515e8fd076f03cb552b64:StgSharp.Common/Mathematics/Linear/MatrixParallel.TaskFactory.cs
+        }
+
+    }
+}