Release collectedRows, Option.bind

Author: kirillgarbar

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

@@ -127,6 +127,7 @@ module SpMSpV =
                 computeOffsetsInplace queue (vector.NNZ * 2 + 1) collectedRows
 
             if gatherArraySize = 0 then
+                collectedRows.Free queue
                 None
             else
                 let ndRange =
@@ -236,39 +237,34 @@ module SpMSpV =
             Reduce.ByKey.Option.segmentSequential add clContext workGroupSize
 
         fun (queue: MailboxProcessor<_>) (matrix: ClMatrix.CSR<'a>) (vector: ClVector.Sparse<'b>) ->
-            let result =
-                gather queue matrix vector
-                |> Option.map
-                    (fun (gatherRows, gatherIndices, gatherValues) ->
-                        sort queue gatherIndices gatherRows gatherValues
-
-                        let sortedRows, sortedIndices, sortedValues = gatherRows, gatherIndices, gatherValues
+            gather queue matrix vector
+            |> Option.map
+                (fun (gatherRows, gatherIndices, gatherValues) ->
+                    sort queue gatherIndices gatherRows gatherValues
 
-                        let multipliedValues =
-                            multiplyScalar queue sortedRows sortedValues vector
+                    let sortedRows, sortedIndices, sortedValues = gatherRows, gatherIndices, gatherValues
 
-                        sortedValues.Free queue
-                        sortedRows.Free queue
+                    let multipliedValues =
+                        multiplyScalar queue sortedRows sortedValues vector
 
-                        let result =
-                            segReduce queue DeviceOnly sortedIndices multipliedValues
-                            |> Option.map
-                                (fun (reducedValues, reducedKeys) ->
+                    sortedValues.Free queue
+                    sortedRows.Free queue
 
-                                    { Context = clContext
-                                      Indices = reducedKeys
-                                      Values = reducedValues
-                                      Size = matrix.ColumnCount })
+                    let result =
+                        segReduce queue DeviceOnly sortedIndices multipliedValues
+                        |> Option.map
+                            (fun (reducedValues, reducedKeys) ->
 
-                        multipliedValues.Free queue
-                        sortedIndices.Free queue
+                                { Context = clContext
+                                  Indices = reducedKeys
+                                  Values = reducedValues
+                                  Size = matrix.ColumnCount })
 
-                        result)
+                    multipliedValues.Free queue
+                    sortedIndices.Free queue
 
-            //Unwrap 't option option to 't option
-            match result with
-            | Some result -> result
-            | None -> None
+                    result)
+            |> Option.bind id
 
 
     let runBoolStandard