SpMSpV general

Author: kirillgarbar

src/GraphBLAS-sharp.Backend/Vector/SpMSpV.fs

@@ -224,6 +224,67 @@ module SpMSpV =
 
             result
 
+    let run
+        (add: Expr<'c option -> 'c option -> 'c option>)
+        (mul: Expr<'a option -> 'b option -> 'c option>)
+        (clContext: ClContext)
+        workGroupSize
+        =
+
+        //TODO: Common.Gather?
+        let gather = gather clContext workGroupSize
+
+        //TODO: Radix sort
+        let sort =
+            Sort.Bitonic.sortKeyValuesInplace clContext workGroupSize
+
+        let multiplyScalar =
+            multiplyScalar clContext mul workGroupSize
+
+        let segReduce =
+            Reduce.ByKey.Option.segmentSequential add clContext workGroupSize
+
+        fun (queue: MailboxProcessor<_>) (matrix: ClMatrix.CSR<'a>) (vector: ClVector.Sparse<'b>) ->
+
+            let gatherRows, gatherIndices, gatherValues, gatherLength = gather queue matrix vector
+
+            if gatherLength <= 0 then
+                gatherRows.Free queue
+                gatherValues.Free queue
+
+                { Context = clContext
+                  Indices = gatherIndices
+                  Values = clContext.CreateClArray 0
+                  Size = matrix.ColumnCount }
+            else
+                sort queue gatherIndices gatherRows gatherValues
+
+                let sortedRows, sortedIndices, sortedValues = gatherRows, gatherIndices, gatherValues
+
+                let multipliedValues =
+                    multiplyScalar queue sortedRows sortedValues vector
+
+                sortedValues.Free queue
+                sortedRows.Free queue
+
+                match segReduce queue DeviceOnly sortedIndices multipliedValues with
+                | Some (reducedValues, reducedKeys) ->
+                    multipliedValues.Free queue
+                    sortedIndices.Free queue
+
+                    { Context = clContext
+                      Indices = reducedKeys
+                      Values = reducedValues
+                      Size = matrix.ColumnCount }
+                | None ->
+                    multipliedValues.Free queue
+                    sortedIndices.Free queue
+
+                    { Context = clContext
+                      Indices = clContext.CreateClArray 0
+                      Values = clContext.CreateClArray 0
+                      Size = matrix.ColumnCount }
+
     let runBoolStandard
         (add: Expr<'c option -> 'c option -> 'c option>)
         (mul: Expr<'a option -> 'b option -> 'c option>)

tests/GraphBLAS-sharp.Tests/Backend/Vector/SpMSpV.fs

@@ -1,5 +1,6 @@
 module GraphBLAS.FSharp.Tests.Backend.Vector.SpMSpV
 
+open System
 open GraphBLAS.FSharp.Backend.Objects.ArraysExtensions
 open Expecto
 open Brahma.FSharp
@@ -131,9 +132,40 @@ let testFixturesSpMSpV (testContext: TestContext) =
           (&&)
           ArithmeticOperations.boolSumOption
           ArithmeticOperations.boolMulOption
-    //createTest testContext 0 1 (=) (+) (*) ArithmeticOperations.intSum ArithmeticOperations.intMul
-    //createTest testContext 0.0f 1f (=) (+) (*) ArithmeticOperations.float32Sum ArithmeticOperations.float32Mul
-     ]
+
+      createTest
+          SpMSpV.run
+          testContext
+          0
+          1
+          (=)
+          (+)
+          (*)
+          ArithmeticOperations.intSumOption
+          ArithmeticOperations.intMulOption
+
+      createTest
+          SpMSpV.run
+          testContext
+          0.0f
+          1f
+          (=)
+          (+)
+          (*)
+          ArithmeticOperations.float32SumOption
+          ArithmeticOperations.float32MulOption
+
+      if Utils.isFloat64Available context.ClDevice then
+          createTest
+              SpMSpV.run
+              testContext
+              0.0
+              1
+              (=)
+              (+)
+              (*)
+              ArithmeticOperations.floatSumOption
+              ArithmeticOperations.floatMulOption ]
 
 let tests =
     gpuTests "Backend.Vector.SpMSpV tests" testFixturesSpMSpV