wip: module Expand test

Author: IgorErin

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

@@ -266,19 +266,19 @@ module Expand =
 
             // since prefix sum include
             // positions in global array for right matrix
-            let globalRightMatrixValuesRawsStartPositions = requiredRawsLengths
+            let globalRightMatrixRawsStartPositions = requiredRawsLengths
 
             // pointers to required raws in right matrix values
             let requiredRightMatrixValuesPointers =
                 getRequiredRightMatrixValuesPointers processor leftMatrix.Columns rightMatrix.RowPointers
 
             // bitmap to distinguish different raws in a general array
             let globalPositions =
-                getGlobalPositions processor globalLength globalRightMatrixValuesRawsStartPositions
+                getGlobalPositions processor globalLength globalRightMatrixRawsStartPositions
 
             // extended pointers to all required right matrix numbers
             let globalRightMatrixValuesPointers =
-                getRightMatrixValuesPointers processor globalLength globalPositions globalRightMatrixValuesRawsStartPositions requiredRightMatrixValuesPointers
+                getRightMatrixValuesPointers processor globalLength globalPositions globalRightMatrixRawsStartPositions requiredRightMatrixValuesPointers
 
             // gather all required right matrix values
             let extendedRightMatrixValues =
@@ -300,6 +300,6 @@ module Expand =
                 map2 processor DeviceOnly extendedLeftMatrixValues extendedRightMatrixValues
 
             let rowPointers =
-                getRawPointers processor leftMatrix.RowPointers globalRightMatrixValuesRawsStartPositions
+                getRawPointers processor leftMatrix.RowPointers globalRightMatrixRawsStartPositions
 
             multiplicationResult, extendedRightMatrixColumns, rowPointers

src/GraphBLAS-sharp.Backend/Objects/ArraysExtentions.fs

@@ -13,6 +13,12 @@ module ArraysExtensions =
             let dst = Array.zeroCreate this.Length
             q.PostAndReply(fun ch -> Msg.CreateToHostMsg(this, dst, ch))
 
+        member this.ToHostAndFree(q: MailboxProcessor<Msg>) =
+            let result = this.ToHost q
+            this.Dispose q
+
+            result
+
         member this.Size = this.Length
 
     type 'a ``[]`` with

tests/GraphBLAS-sharp.Tests/Matrix/SpGEMM/Expand.fs

@@ -5,9 +5,15 @@ open GraphBLAS.FSharp.Backend.Matrix.CSR.SpGEMM
 open GraphBLAS.FSharp.Tests
 open GraphBLAS.FSharp.Backend.Objects.ArraysExtensions
 open Expecto
+open GraphBLAS.FSharp.Backend.Common
+open GraphBLAS.FSharp.Backend.Predefined
+open GraphBLAS.FSharp.Backend.Objects.ClCell
 
 let context = Context.defaultContext
 
+let clContext = context.ClContext
+let processor = context.Queue
+
 /// <remarks>
 /// Left matrix
 /// </remarks>
@@ -42,25 +48,94 @@ let rightMatrix =
       ColumnIndices = [| 1; 4; 6; 2; 5; 1; 5; 6; 4; 6 |]
       Values = [| 3; 4; 4; 2; 2; 5; 9; 1; 1; 8 |] }
 
-let requiredRowLength =
+let deviceLeftMatrix = leftMatrix.ToDevice clContext
+let deviceRightMatrix = rightMatrix.ToDevice clContext
+
+let requiredRawsLengths () =
+    let getRequiredRawsLengths =
+        Expand.processLeftMatrixColumnsAndRightMatrixRawPointers clContext Utils.defaultWorkGroupSize Expand.requiredRawsLengths
+
+    getRequiredRawsLengths processor deviceLeftMatrix.Columns deviceRightMatrix.RowPointers
+
+let requiredRowLengthTest =
     testCase "requiredRowLength"
     <| fun () ->
-        let clContext = context.ClContext
-        let processor = context.Queue
+        let actual = requiredRawsLengths().ToHostAndFree processor
+
+        "Results must be the same"
+        |> Expect.equal actual [| 2; 3; 3; 3; 2; 2; 0; 3 |]
+
+let globalLength =
+    let prefixSumExclude =
+        PrefixSum.standardExcludeInplace clContext Utils.defaultWorkGroupSize
 
-        let deviceLeftMatrix = leftMatrix.ToDevice clContext
-        let deviceRightMatrix = rightMatrix.ToDevice clContext
+    let requiredRawsLengths = requiredRawsLengths ()
 
-        let getRequiredRawsLengths =
-            Expand.processLeftMatrixColumnsAndRightMatrixRawPointers clContext Utils.defaultWorkGroupSize Expand.requiredRawsLengths
+    (prefixSumExclude processor requiredRawsLengths).ToHostAndFree processor
 
-        let requiredRawsLengths =
-            getRequiredRawsLengths processor deviceLeftMatrix.Columns deviceRightMatrix.RowPointers
+let globalLengthTest =
+    testCase "global length test"
+    <| fun () -> Expect.equal globalLength 18 "Results must be the same"
 
-        let requiredRawsLengthsHost = requiredRawsLengths.ToHost processor
+let getGlobalRightMatrixRawsStartPositions () =
+    let prefixSumExclude =
+        PrefixSum.standardExcludeInplace clContext Utils.defaultWorkGroupSize
+
+    let requiredRawsLengths = requiredRawsLengths ()
+
+    (prefixSumExclude processor requiredRawsLengths).Free processor
+
+    requiredRawsLengths
+
+let globalRightMatrixRawsStartPositionsTest =
+    testCase "global right matrix raws start positions"
+    <| fun () ->
+        let result = (getGlobalRightMatrixRawsStartPositions ()).ToHostAndFree processor
 
         "Results must be the same"
-        |> Expect.equal requiredRawsLengthsHost [| 2; 3; 3; 3; 2; 2; 0; 3 |]
+        |> Expect.equal result [| 0; 2; 5; 8; 11; 13; 15; 15; |]
+
+let getRequiredRightMatrixValuesPointers () =
+    let getRequiredRightMatrixValuesPointers =
+        Expand.processLeftMatrixColumnsAndRightMatrixRawPointers clContext Utils.defaultWorkGroupSize Expand.requiredRawPointers
 
+    getRequiredRightMatrixValuesPointers processor deviceLeftMatrix.Columns deviceRightMatrix.RowPointers
 
+let getRequiredRightMatrixValuesPointersTest =
+    testCase "get required right matrix values pointers"
+    <| fun () ->
+        let result = (getRequiredRightMatrixValuesPointers ()).ToHostAndFree processor
+
+        "Result must be the same"
+        |> Expect.equal result [| 3; 5; 0; 5; 8; 3; 0; 0; |]
+
+let getGlobalPositions () =
+    let getGlobalPositions = Expand.getGlobalPositions clContext Utils.defaultWorkGroupSize
+
+    getGlobalPositions processor globalLength (getGlobalRightMatrixRawsStartPositions ())
+
+let getGlobalPositionsTest =
+    testCase "getGlobalPositions test"
+    <| fun () ->
+        let result = (getGlobalPositions ()).ToHostAndFree processor
+
+        "Result must be the same"
+        |> Expect.equal result [| 1; 1; 2; 2; 2; 3; 3; 3; 4; 4; 4; 5; 5; 6; 6; 7; 7; 7; |]
+
+let getRightMatrixValuesPointers () =
+    let getRightMatrixValuesPointers =
+        Expand.getRightMatrixPointers clContext Utils.defaultWorkGroupSize
+
+    let globalPositions = getGlobalPositions ()
+    let globalRightMatrixRawsStartPositions = getGlobalRightMatrixRawsStartPositions ()
+    let requiredRightMatrixValuesPointers = getRequiredRightMatrixValuesPointers ()
+
+    getRightMatrixValuesPointers processor globalLength globalPositions globalRightMatrixRawsStartPositions requiredRightMatrixValuesPointers
+
+let rightMatrixValuesPointersTest =
+    testCase "RightMatrixValuesPointers"
+    <| fun () ->
+        let result = (getRightMatrixValuesPointers ()).ToHostAndFree processor
 
+        "Result must be the same"
+        |> Expect.equal result [| 3; 4; 5; 6; 7; 0; 1; 2; 5; 6; 7; 8; 9; 3; 4; 0; 1; 2; |]

tests/GraphBLAS-sharp.Tests/Program.fs

@@ -64,7 +64,12 @@ open GraphBLAS.FSharp.Tests.Backend
 let allTests =
     testList
         "All tests"
-        [ Matrix.SpGEMM.Expand.requiredRowLength ]
+        [ Matrix.SpGEMM.Expand.requiredRowLengthTest
+          Matrix.SpGEMM.Expand.globalLengthTest
+          Matrix.SpGEMM.Expand.globalRightMatrixRawsStartPositionsTest
+          Matrix.SpGEMM.Expand.getRequiredRightMatrixValuesPointersTest
+          Matrix.SpGEMM.Expand.getGlobalPositionsTest
+          Matrix.SpGEMM.Expand.rightMatrixValuesPointersTest ]
     |> testSequenced
 
 [<EntryPoint>]