add: matrix intersect

Author: artemiipatov

src/GraphBLAS-sharp.Backend/GraphBLAS-sharp.Backend.fsproj

@@ -53,6 +53,7 @@
     <Compile Include="Matrix/COO/Map.fs" />
     <Compile Include="Matrix/COO/Merge.fs" />
     <Compile Include="Matrix/COO/Map2.fs" />
+    <Compile Include="Matrix/COO/Intersect.fs" />
     <Compile Include="Matrix/COO/Matrix.fs" />
     <Compile Include="Matrix/CSR/Merge.fs" />
     <Compile Include="Matrix/CSR/Map2.fs" />

src/GraphBLAS-sharp.Backend/Matrix/COO/Intersect.fs

@@ -0,0 +1,64 @@
+namespace GraphBLAS.FSharp.Backend.Matrix.COO
+
+open Brahma.FSharp
+open GraphBLAS.FSharp
+open GraphBLAS.FSharp.Objects
+open GraphBLAS.FSharp.Backend.Quotes
+open GraphBLAS.FSharp.Objects.ClContextExtensions
+
+module internal Intersect =
+    let findIntersectionByKeys (clContext: ClContext) workGroupSize =
+
+        let findIntersection =
+            <@ fun (ndRange: Range1D) (leftNNZ: int) (rightNNZ: int) (leftRows: ClArray<int>) (leftColumns: ClArray<int>) (rightRows: ClArray<int>) (rightColumns: ClArray<int>) (bitmap: ClArray<int>) ->
+
+                let gid = ndRange.GlobalID0
+                let bitmapSize = min leftNNZ rightNNZ
+
+                if gid < bitmapSize then
+
+                    let row, column, rows, columns =
+                        if rightNNZ >= leftNNZ then
+                            leftRows.[gid], leftColumns.[gid], leftRows, leftColumns
+                        else
+                            rightRows.[gid], rightColumns.[gid], rightRows, rightColumns
+
+                    let index: uint64 = ((uint64 row) <<< 32) ||| (uint64 column)
+
+                    let intersect = (%Search.Bin.existsByKey2D) bitmapSize index rows columns
+
+                    if intersect then
+                        bitmap.[gid] <- 1
+                    else
+                        bitmap.[gid] <- 0 @>
+
+        let kernel =
+            clContext.Compile <| findIntersection
+
+        fun (processor: MailboxProcessor<_>) allocationMode (leftMatrix: ClMatrix.COO<'a>) (rightMatrix: ClMatrix.COO<'b>) ->
+
+            let bitmapSize = min leftMatrix.NNZ rightMatrix.NNZ
+
+            let bitmap = clContext.CreateClArrayWithSpecificAllocationMode(allocationMode, bitmapSize)
+
+            let ndRange = Range1D.CreateValid(bitmapSize, workGroupSize)
+
+            let kernel = kernel.GetKernel()
+
+            processor.Post(
+                Msg.MsgSetArguments
+                    (fun () ->
+                        kernel.KernelFunc
+                            ndRange
+                            leftMatrix.NNZ
+                            rightMatrix.NNZ
+                            leftMatrix.Rows
+                            leftMatrix.Columns
+                            rightMatrix.Rows
+                            rightMatrix.Columns
+                            bitmap)
+            )
+
+            processor.Post(Msg.CreateRunMsg<_, _> kernel)
+
+            bitmap

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

@@ -197,8 +197,8 @@ module Matrix =
     /// <summary>
     /// Transposes the given matrix and returns result as a new matrix.
     /// </summary>
-    ///<param name="clContext">OpenCL context.</param>
-    ///<param name="workGroupSize">Should be a power of 2 and greater than 1.</param>
+    /// <param name="clContext">OpenCL context.</param>
+    /// <param name="workGroupSize">Should be a power of 2 and greater than 1.</param>
     let transpose (clContext: ClContext) workGroupSize =
 
         let transposeInPlace = transposeInPlace clContext workGroupSize
@@ -216,3 +216,12 @@ module Matrix =
               Columns = copy queue allocationMode matrix.Columns
               Values = copyData queue allocationMode matrix.Values }
             |> transposeInPlace queue
+
+    /// <summary>
+    /// Build a bitmap. Maps non-zero elements of the matrix with minimum nnz to 1
+    /// if the second matrix has non zero element under the same row and column pair, otherwise 0.
+    /// </summary>
+    /// <param name="clContext">OpenCL context.</param>
+    /// <param name="workGroupSize">Should be a power of 2 and greater than 1.</param>
+    let findIntersectionByKeys (clContext: ClContext) workGroupSize =
+        Intersect.findIntersectionByKeys clContext workGroupSize

src/GraphBLAS-sharp.Backend/Quotes/Search.fs

@@ -96,6 +96,36 @@ module Search =
 
                 result @>
 
+        /// <summary>
+        /// Searches value in array by two keys.
+        /// In case there is a value at the given keys position, it is returned.
+        /// </summary>
+        let existsByKey2D<'a> =
+            <@ fun length sourceIndex (rowIndices: ClArray<int>) (columnIndices: ClArray<int>) ->
+
+                let mutable leftEdge = 0
+                let mutable rightEdge = length - 1
+
+                let mutable result = false
+
+                while leftEdge <= rightEdge do
+                    let middleIdx = (leftEdge + rightEdge) / 2
+
+                    let currentIndex: uint64 =
+                        ((uint64 rowIndices.[middleIdx]) <<< 32)
+                        ||| (uint64 columnIndices.[middleIdx])
+
+                    if sourceIndex = currentIndex then
+                        result <- true
+
+                        rightEdge <- -1 // TODO() break
+                    elif sourceIndex < currentIndex then
+                        rightEdge <- middleIdx - 1
+                    else
+                        leftEdge <- middleIdx + 1
+
+                result @>
+
         /// <summary>
         /// Find lower position of item in array.
         /// </summary>