add: Split

Author: IgorErin

src/GraphBLAS-sharp.Backend/Matrix/Split.fs

@@ -2,21 +2,74 @@ module GraphBLAS.FSharp.Backend.Matrix
 
 open Brahma.FSharp
 open GraphBLAS.FSharp.Backend.Common
-open GraphBLAS.FSharp.Backend.Matrix
-open GraphBLAS.FSharp.Backend.Quotes
-open Microsoft.FSharp.Quotations
 open GraphBLAS.FSharp.Backend.Objects
 open GraphBLAS.FSharp.Backend.Objects.ClMatrix
-open GraphBLAS.FSharp.Backend.Objects.ClContext
+open GraphBLAS.FSharp.Backend.Objects.ArraysExtensions
+
+// type lazy matrix ???
 
 module Split =
-    let toCOO (clContext: ClContext) workGroupSize =
+    module ByChunk =
+        let runCOOLazy (clContext: ClContext) workGroupSize =
+
+            let chunkBySizeValues = ClArray.lazyChunkBySize clContext workGroupSize
+
+            let chunkBySizeIndices = ClArray.lazyChunkBySize clContext workGroupSize
+
+            fun (processor: MailboxProcessor<_>) allocationMode chunkSize (matrix: ClMatrix.COO<'a>) ->
+
+               let createSubMatrixLazy (values: Lazy<_>) (columns: Lazy<_>) (rows: Lazy<_>) =
+                   lazy
+                   { Context = clContext
+                     RowCount = matrix.RowCount
+                     ColumnCount = matrix.ColumnCount
+                     Rows = rows.Value
+                     Columns = columns.Value
+                     Values = values.Value }
+
+               let values = chunkBySizeValues processor allocationMode chunkSize matrix.Values
+               let columns = chunkBySizeIndices processor allocationMode chunkSize matrix.Columns
+               let rows = chunkBySizeIndices processor allocationMode chunkSize matrix.Rows
+
+               Seq.map3 createSubMatrixLazy values columns rows
+
+        let runCOO (clContext: ClContext) workGroupSize =
+
+            let run = runCOOLazy clContext workGroupSize
+
+            fun (processor: MailboxProcessor<_>) allocationMode chunkSize (matrix: ClMatrix.COO<'a>) ->
+               run processor allocationMode chunkSize matrix
+               |> Seq.map (fun lazyMatrix -> lazyMatrix.Value)
+               |> Seq.toArray
+
+    module ByRow =
+        let runCSRLazy (clContext: ClContext) workGroupSize =
+
+            let getChunkValues = ClArray.getChunk clContext workGroupSize
+
+            let getChunkIndices = ClArray.getChunk clContext workGroupSize
+
+            fun (processor: MailboxProcessor<_>) allocationMode (matrix: ClMatrix.CSR<'a>) ->
+
+                let getChunkValues = getChunkValues processor allocationMode matrix.Values
+                let getChunkIndices = getChunkIndices processor allocationMode matrix.Columns
+
+                matrix.RowPointers.ToHost processor
+                |> Seq.pairwise
+                |> Seq.map (fun (first, second) ->
+                    lazy
+                        if second - first > 0 then
+                            let values = getChunkValues first second
+                            let columns = getChunkIndices first second
 
-        let copy = ClArray.copy clContext workGroupSize
+                            Some (values, columns)
+                        else None)
 
-        let copyData = ClArray.copy clContext workGroupSize
+        let runCSR (clContext: ClContext) workGroupSize =
 
-        // endIndex exclusive (for csr matrix row pointers interop), startIndex inclusive
-        fun (processor: MailboxProcessor<_>) allocationMode (matrix: ClMatrix.CSR<'a>) startIndex endIndex ->
+            let runLazy = runCSRLazy clContext workGroupSize
 
-            ()
+            fun (processor: MailboxProcessor<_>) allocationMode (matrix: ClMatrix.CSR<'a>) ->
+                runLazy processor allocationMode matrix
+                |> Seq.map (fun lazyValue -> lazyValue.Value)
+                |> Seq.toArray