Method for array creation with flags

Author: kirillgarbar

src/GraphBLAS-sharp.Backend/Common/ClArray.fs

@@ -1,11 +1,27 @@
 namespace GraphBLAS.FSharp.Backend
 
-open Brahma.FSharp.OpenCL
-open GraphBLAS.FSharp.Backend.Common
+open GraphBLAS.FSharp.Backend
 open Brahma.FSharp
 open Microsoft.FSharp.Quotations
 
 module ClArray =
+    type ClContext with
+        member this.CreateClArrayWithGPUOnlyFlags(size: int) =
+            this.CreateClArray(
+                size,
+                deviceAccessMode = DeviceAccessMode.ReadWrite,
+                hostAccessMode = HostAccessMode.NotAccessible,
+                allocationMode = AllocationMode.Default
+            )
+
+        member this.CreateClArrayWithGPUOnlyFlags(array: 'a []) =
+            this.CreateClArray(
+                array,
+                deviceAccessMode = DeviceAccessMode.ReadWrite,
+                hostAccessMode = HostAccessMode.NotAccessible,
+                allocationMode = AllocationMode.CopyHostPtr
+            )
+
     let init (initializer: Expr<int -> 'a>) (clContext: ClContext) workGroupSize =
 
         let init =
@@ -20,7 +36,8 @@ module ClArray =
 
         fun (processor: MailboxProcessor<_>) (length: int) ->
             // TODO: Выставить нужные флаги
-            let outputArray = clContext.CreateClArray(length)
+            let outputArray =
+                clContext.CreateClArrayWithGPUOnlyFlags(length)
 
             let kernel = program.GetKernel()
 
@@ -47,7 +64,8 @@ module ClArray =
         fun (processor: MailboxProcessor<_>) (length: int) (value: 'a) ->
             let value = clContext.CreateClCell(value)
 
-            let outputArray = clContext.CreateClArray(length)
+            let outputArray =
+                clContext.CreateClArrayWithGPUOnlyFlags(length)
 
             let kernel = program.GetKernel()
 
@@ -83,7 +101,7 @@ module ClArray =
                 Range1D.CreateValid(inputArray.Length, workGroupSize)
 
             let outputArray =
-                clContext.CreateClArray(inputArray.Length, allocationMode = AllocationMode.Default)
+                clContext.CreateClArrayWithGPUOnlyFlags inputArray.Length
 
             let kernel = program.GetKernel()
 
@@ -111,7 +129,7 @@ module ClArray =
             let outputArrayLength = inputArray.Length * count
 
             let outputArray =
-                clContext.CreateClArray(outputArrayLength, allocationMode = AllocationMode.Default)
+                clContext.CreateClArrayWithGPUOnlyFlags outputArrayLength
 
             let ndRange =
                 Range1D.CreateValid(outputArray.Length, workGroupSize)
@@ -228,12 +246,7 @@ module ClArray =
                 Range1D.CreateValid(inputLength, workGroupSize)
 
             let bitmap =
-                clContext.CreateClArray(
-                    inputLength,
-                    hostAccessMode = HostAccessMode.NotAccessible,
-                    deviceAccessMode = DeviceAccessMode.ReadWrite,
-                    allocationMode = AllocationMode.Default
-                )
+                clContext.CreateClArrayWithGPUOnlyFlags inputLength
 
             let kernel = kernel.GetKernel()
 
@@ -261,7 +274,7 @@ module ClArray =
                 Range1D.CreateValid(inputArray.Length, workGroupSize)
 
             let outputArray =
-                clContext.CreateClArray(outputArraySize, allocationMode = AllocationMode.Default)
+                clContext.CreateClArrayWithGPUOnlyFlags outputArraySize
 
             let kernel = kernel.GetKernel()
 

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

@@ -7,6 +7,7 @@ open Microsoft.FSharp.Quotations
 open GraphBLAS.FSharp.Backend.Common
 open GraphBLAS.FSharp.Backend.DenseVector
 open GraphBLAS.FSharp.Backend.SparseVector
+open GraphBLAS.FSharp.Backend.ClArray
 
 module Vector =
     let zeroCreate (clContext: ClContext) (workGroupSize: int) =
@@ -18,8 +19,8 @@ module Vector =
             | Sparse ->
                 let vector =
                     { Context = clContext
-                      Indices = clContext.CreateClArray [| 0 |]
-                      Values = clContext.CreateClArray [| Unchecked.defaultof<'a> |]
+                      Indices = clContext.CreateClArrayWithGPUOnlyFlags [| 0 |]
+                      Values = clContext.CreateClArrayWithGPUOnlyFlags [| Unchecked.defaultof<'a> |]
                       Size = size }
 
                 ClVectorSparse vector
@@ -45,7 +46,7 @@ module Vector =
                 for i in 0 .. indices.Length - 1 do
                     res.[indices.[i]] <- Some(values.[i])
 
-                ClVectorDense <| clContext.CreateClArray res
+                ClVectorDense <| clContext.CreateClArrayWithGPUOnlyFlags res
 
     let copy (clContext: ClContext) (workGroupSize: int) =
         let copy = ClArray.copy clContext workGroupSize