refactor: wip

Author: IgorErin

src/GraphBLAS-sharp.Backend/Common/ClArray.fs

@@ -5,6 +5,7 @@ open Microsoft.FSharp.Quotations
 open GraphBLAS.FSharp.Backend.Objects.ClContext
 open GraphBLAS.FSharp.Backend.Objects.ClCell
 open GraphBLAS.FSharp.Backend.Quotes
+open GraphBLAS.FSharp.Backend.Objects.ArraysExtensions
 
 module ClArray =
     let init (clContext: ClContext) workGroupSize (initializer: Expr<int -> 'a>) =
@@ -33,30 +34,6 @@ module ClArray =
 
             outputArray
 
-    let assignManyInit (clContext: ClContext) workGroupSize (initializer: Expr<int -> 'a>) =
-
-        let init =
-            <@ fun (range: Range1D) indicesLength (indices: ClArray<int>) (outputBuffer: ClArray<'a>) ->
-
-                let gid = range.GlobalID0
-
-                if gid < indicesLength then
-                    let targetIndex = indices.[gid]
-
-                    outputBuffer.[targetIndex] <- (%initializer) gid @>
-
-        let program = clContext.Compile(init)
-
-        fun (processor: MailboxProcessor<_>) (indices: ClArray<int>) (result: ClArray<'a>) ->
-
-            let kernel = program.GetKernel()
-
-            let ndRange =
-                Range1D.CreateValid(indices.Length, workGroupSize)
-
-            processor.Post(Msg.MsgSetArguments(fun () -> kernel.KernelFunc ndRange indices.Length indices result))
-            processor.Post(Msg.CreateRunMsg<_, _> kernel)
-
     let create (clContext: ClContext) workGroupSize =
 
         let create =
@@ -315,6 +292,24 @@ module ClArray =
 
             resultArray
 
+    let getUniqueBitmap2<'a when 'a: equality> (clContext: ClContext) workGroupSize =
+
+        let map = map2 clContext workGroupSize <@ fun x y -> if x = 1 && y = 1 then 1 else 0 @>
+
+        let getUniqueBitmap = getUniqueBitmap clContext workGroupSize
+
+        fun (processor: MailboxProcessor<_>) allocationMode (firstArray: ClArray<'a>) (secondArray: ClArray<'a>) ->
+            let firstBitmap = getUniqueBitmap processor DeviceOnly firstArray
+
+            let secondBitmap = getUniqueBitmap processor DeviceOnly secondArray
+
+            let result = map processor allocationMode firstBitmap secondBitmap
+
+            firstBitmap.Free processor
+            secondBitmap.Free processor
+
+            result
+
     let choose<'a, 'b> (clContext: ClContext) workGroupSize (predicate: Expr<'a -> 'b option>) =
         let getBitmap =
             map<'a, int> clContext workGroupSize
@@ -353,3 +348,14 @@ module ClArray =
             scatter processor positions values result
 
             result
+
+    let iterate (clContext: ClContext) workGroupSize iterator =
+
+        let create = create clContext workGroupSize iterator
+
+        let scatter = Scatter.runInplace clContext workGroupSize
+
+        fun (processor: MailboxProcessor<_>) allocationMode (inputArray: ClArray<'a>) (resultArray: ClArray<'a>) ->
+
+            let positions = create processor allocationMode
+

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

@@ -9,6 +9,7 @@ open GraphBLAS.FSharp.Backend.Objects
 open GraphBLAS.FSharp.Backend.Objects.ClMatrix
 open GraphBLAS.FSharp.Backend.Objects.ClContext
 open GraphBLAS.FSharp.Backend.Objects.ArraysExtensions
+open GraphBLAS.FSharp.Backend.Objects.ClCell
 
 module Matrix =
     let private expandRowPointers (clContext: ClContext) workGroupSize =
@@ -155,50 +156,3 @@ module Matrix =
 
             run queue matrixLeft matrixRight mask
 
-    let spgemm
-        (clContext: ClContext)
-        workGroupSize
-        (opAdd: Expr<'c -> 'c -> 'c>)
-        (opMul: Expr<'a -> 'b -> 'c>)
-        =
-
-        let expand = SpGEMM.Expand.run clContext workGroupSize opMul
-
-        let expandRowPointers = expandRowPointers clContext workGroupSize
-
-        let sortData = Sort.Radix.runByKeysStandard clContext workGroupSize
-
-        let sortKeys = Sort.Radix.runByKeysStandard clContext workGroupSize
-
-        let reduceByKey = Reduce.ByKey.segmentSequential clContext workGroupSize opAdd
-
-        fun (processor: MailboxProcessor<_>) (leftMatrix: ClMatrix.CSR<'a>) (rightMatrix: ClMatrix.CSR<'b>) ->
-            let multiplicationResult, columns, rowPointers =
-                expand processor leftMatrix rightMatrix
-
-            let rows =
-                expandRowPointers processor DeviceOnly rowPointers columns.Length leftMatrix.RowCount
-
-            rowPointers.Free processor
-
-            // sorting
-            let sortData = sortData processor
-            let sortKeys = sortKeys processor
-
-            // by columns
-            let valuesSortedByColumns = sortData columns multiplicationResult
-            let byKeSortedRows = sortKeys columns rows
-
-            multiplicationResult.Free processor
-            rows.Free processor
-
-            // by rows
-            let values = sortData byKeSortedRows valuesSortedByColumns
-            let columns = sortKeys byKeSortedRows columns
-
-            // reduce
-
-
-
-            ()
-