PageRankMatrix type

Author: kirillgarbar

benchmarks/GraphBLAS-sharp.Benchmarks/Algorithms/PageRank.fs

@@ -23,7 +23,7 @@ type Benchmarks(
 
     let mutable funToBenchmark = None
     let mutable matrix = Unchecked.defaultof<ClMatrix<float32>>
-    let mutable matrixPrepared = Unchecked.defaultof<ClMatrix<float32>>
+    let mutable matrixPrepared = Unchecked.defaultof<PageRankMatrix<float32>>
     let mutable matrixHost = Unchecked.defaultof<_>
 
     let accuracy = 0.00000001f

src/GraphBLAS-sharp.Backend/Algorithms/PageRank.fs

@@ -131,6 +131,7 @@ module internal PageRank =
 
                 transposeInPlace queue DeviceOnly newMatrix
                 |> ClMatrix.CSR
+                |> PageRankMatrix
             | _ -> failwith "Not implemented"
 
     let run (clContext: ClContext) workGroupSize =
@@ -154,8 +155,7 @@ module internal PageRank =
         let create =
             GraphBLAS.FSharp.Vector.create clContext workGroupSize
 
-        fun (queue: MailboxProcessor<Msg>) (matrix: ClMatrix<float32>) accuracy ->
-
+        fun (queue: MailboxProcessor<Msg>) (PageRankMatrix matrix) accuracy ->
             let vertexCount = matrix.RowCount
 
             //None is 0

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

@@ -173,3 +173,36 @@ type ClMatrix<'a when 'a: struct> =
         | ClMatrix.COO matrix -> matrix.NNZ
         | ClMatrix.CSC matrix -> matrix.NNZ
         | ClMatrix.LIL matrix -> matrix.NNZ
+
+/// <summary>
+/// Represents an abstraction over matrix, which is converted to correct format for PageRank algorithm
+/// </summary>
+type PageRankMatrix<'a when 'a: struct> =
+    | PageRankMatrix of ClMatrix<'a>
+    /// <summary>
+    /// Gets the number of rows in matrix.
+    /// </summary>
+    member this.RowCount =
+        match this with
+        | PageRankMatrix matrix -> matrix.RowCount
+
+    /// <summary>
+    /// Gets the number of columns in matrix.
+    /// </summary>
+    member this.ColumnCount =
+        match this with
+        | PageRankMatrix matrix -> matrix.ColumnCount
+
+    /// <summary>
+    /// Release device resources allocated for the matrix.
+    /// </summary>
+    member this.Dispose q =
+        match this with
+        | PageRankMatrix matrix -> matrix.Dispose q
+
+    /// <summary>
+    /// Gets the number of non-zero elements in matrix.
+    /// </summary>
+    member this.NNZ =
+        match this with
+        | PageRankMatrix matrix -> matrix.NNZ

tests/GraphBLAS-sharp.Tests/Program.fs

@@ -92,7 +92,8 @@ let algorithmsTests =
     testList
         "Algorithms tests"
         [ Algorithms.BFS.tests
-          Algorithms.SSSP.tests ]
+          Algorithms.SSSP.tests
+          Algorithms.PageRank.tests ]
     |> testSequenced
 
 let deviceTests =