refactor: benchmarks

Author: IgorErin

benchmarks/GraphBLAS-sharp.Benchmarks/BenchmarksBFS.fs renamed to benchmarks/GraphBLAS-sharp.Benchmarks/Algorithms/BFS.fs

@@ -1,43 +1,41 @@
 namespace GraphBLAS.FSharp.Benchmarks
 
 open System.IO
+open BenchmarkDotNet.Attributes
+open GraphBLAS.FSharp
 open GraphBLAS.FSharp.Backend.Quotes
 open GraphBLAS.FSharp.IO
-open BenchmarkDotNet.Attributes
-open BenchmarkDotNet.Configs
-open BenchmarkDotNet.Columns
 open Brahma.FSharp
-open GraphBLAS.FSharp.Objects
+open Backend.Algorithms.BFS
+open Microsoft.FSharp.Core
+open GraphBLAS.FSharp.Backend.Objects.ArraysExtensions
+open GraphBLAS.FSharp.Benchmarks
 open GraphBLAS.FSharp.Backend.Objects
-open GraphBLAS.FSharp.Backend.Algorithms
-open MatrixExtensions
-open ArraysExtensions
 
 [<AbstractClass>]
-[<IterationCount(10)>]
-[<WarmupCount(5)>]
-[<Config(typeof<AlgorithmConfig>)>]
-type BFSBenchmarks<'matrixT, 'elem when 'matrixT :> IDeviceMemObject and 'elem : struct>(
- buildFunToBenchmark,
- converter: string -> 'elem,
- converterBool,
- buildMatrix) =
+[<IterationCount(100)>]
+[<WarmupCount(10)>]
+[<Config(typeof<Configs.SingleMatrixConfig>)>]
+type BFSBenchmarks<'elem when 'elem : struct>(
+ buildFunToBenchmark,
+ converter: string -> 'elem,
+ binaryConverter,
+ vertex: int)
+ =
 
  let mutable funToBenchmark = None
- let mutable matrix = Unchecked.defaultof<'matrixT>
+ let mutable matrix = Unchecked.defaultof<ClMatrix.CSR<'elem>>
  let mutable matrixHost = Unchecked.defaultof<_>
 
- let source = 0
-
- member val ResultVector = Unchecked.defaultof<ClArray<'elem option>> with get,set
+ member val ResultLevels = Unchecked.defaultof<ClArray<'elem option>> with get,set
 
  [<ParamsSource("AvailableContexts")>]
  member val OclContextInfo = Unchecked.defaultof<Utils.BenchmarkContext * int> with get, set
 
- [<ParamsSource("InputMatricesProvider")>]
+ [<ParamsSource("InputMatrixProvider")>]
  member val InputMatrixReader = Unchecked.defaultof<MtxReader> with get, set
 
- member this.OclContext:ClContext = (fst this.OclContextInfo).ClContext
+ member this.OclContext = (fst this.OclContextInfo).ClContext
  member this.WorkGroupSize = snd this.OclContextInfo
 
  member this.Processor =
@@ -47,17 +45,16 @@ type BFSBenchmarks<'matrixT, 'elem when 'matrixT :> IDeviceMemObject and 'elem :
 
  static member AvailableContexts = Utils.avaliableContexts
 
- static member InputMatricesProviderBuilder pathToConfig =
- let datasetFolder = ""
+ static member InputMatrixProviderBuilder pathToConfig =
+ let datasetFolder = "BFS"
  pathToConfig
  |> Utils.getMatricesFilenames
  |> Seq.map
  (fun matrixFilename ->
  printfn "%A" matrixFilename
 
  match Path.GetExtension matrixFilename with
- | ".mtx" ->
- MtxReader(Utils.getFullPathToMatrix datasetFolder matrixFilename)
+ | ".mtx" -> MtxReader(Utils.getFullPathToMatrix datasetFolder matrixFilename)
  | _ -> failwith "Unsupported matrix format")
 
  member this.FunToBenchmark =
@@ -68,29 +65,24 @@ type BFSBenchmarks<'matrixT, 'elem when 'matrixT :> IDeviceMemObject and 'elem :
  x
  | Some x -> x
 
- member this.ReadMatrix (reader:MtxReader) =
- let converter =
- match reader.Field with
- | Pattern -> converterBool
- | _ -> converter
-
- reader.ReadMatrix converter
-
  member this.BFS() =
- this.ResultVector <- this.FunToBenchmark this.Processor matrix source
+ this.ResultLevels <- this.FunToBenchmark this.Processor matrix vertex
 
  member this.ClearInputMatrix() =
  (matrix :> IDeviceMemObject).Dispose this.Processor
 
- member this.ClearResult() =
- this.ResultVector.FreeAndWait this.Processor
+ member this.ClearResult() = this.ResultLevels.FreeAndWait this.Processor
 
  member this.ReadMatrix() =
- let matrixReader = this.InputMatrixReader
- matrixHost <- this.ReadMatrix matrixReader
+ let converter =
+ match this.InputMatrixReader.Field with
+ | Pattern -> binaryConverter
+ | _ -> converter
+
+ matrixHost <- this.InputMatrixReader.ReadMatrix converter
 
  member this.LoadMatrixToGPU() =
- matrix <- buildMatrix this.OclContext matrixHost
+ matrix <- matrixHost.ToCSR.ToDevice this.OclContext
 
  abstract member GlobalSetup : unit -> unit
 
@@ -100,21 +92,22 @@ type BFSBenchmarks<'matrixT, 'elem when 'matrixT :> IDeviceMemObject and 'elem :
 
  abstract member Benchmark : unit -> unit
 
-type BFSBenchmarksWithoutDataTransfer() =
+type BFSBenchmarksWithoutDataTransfer<'elem when 'elem : struct>(
+ buildFunToBenchmark,
+ converter: string -> 'elem,
+ boolConverter,
+ vertex) =
 
- inherit BFSBenchmarks<ClMatrix.CSR<int>, int>(
- (fun context wgSize -> BFS.singleSource context ArithmeticOperations.intSumOption ArithmeticOperations.intMulOption wgSize),
- int,
- (fun _ -> Utils.nextInt (System.Random())),
- Matrix.ToBackendCSR)
-
- static member InputMatricesProvider =
- BFSBenchmarks<_,_>.InputMatricesProviderBuilder "BFSBenchmarks.txt"
+ inherit BFSBenchmarks<'elem>(
+ buildFunToBenchmark,
+ converter,
+ boolConverter,
+ vertex)
 
  [<GlobalSetup>]
  override this.GlobalSetup() =
- this.ReadMatrix ()
- this.LoadMatrixToGPU ()
+ this.ReadMatrix()
+ this.LoadMatrixToGPU()
 
  [<IterationCleanup>]
  override this.IterationCleanup() =
@@ -127,27 +120,27 @@ type BFSBenchmarksWithoutDataTransfer() =
  [<Benchmark>]
  override this.Benchmark() =
  this.BFS()
- this.Processor.PostAndReply(Msg.MsgNotifyMe)
+ this.Processor.PostAndReply Msg.MsgNotifyMe
 
-type BFSBenchmarksWithDataTransfer<'matrixT, 'elem when 'matrixT :> IDeviceMemObject and 'elem : struct>(
- buildFunToBenchmark,
- converter: string -> 'elem,
- converterBool,
- buildMatrix,
- resultToHost) =
+type BFSBenchmarksWithTransfer<'elem when 'elem : struct>(
+ buildFunToBenchmark,
+ converter: string -> 'elem,
+ boolConverter,
+ vertex) =
 
- inherit BFSBenchmarks<'matrixT, 'elem>(
+ inherit BFSBenchmarks<'elem>(
  buildFunToBenchmark,
  converter,
- converterBool,
- buildMatrix)
+ boolConverter,
+ vertex)
 
  [<GlobalSetup>]
  override this.GlobalSetup() =
  this.ReadMatrix()
 
  [<GlobalCleanup>]
- override this.GlobalCleanup() = ()
+ override this.GlobalCleanup() =
+ this.ClearResult()
 
  [<IterationCleanup>]
  override this.IterationCleanup() =
@@ -158,7 +151,28 @@ type BFSBenchmarksWithDataTransfer<'matrixT, 'elem when 'matrixT :> IDeviceMemOb
  override this.Benchmark() =
  this.LoadMatrixToGPU()
  this.BFS()
+ this.ResultLevels.ToHost this.Processor |> ignore
  this.Processor.PostAndReply Msg.MsgNotifyMe
- resultToHost this.ResultVector this.Processor
- this.Processor.PostAndReply Msg.MsgNotifyMe
+
+type BFSIntWithoutTransferBenchmark() =
+
+ inherit BFSBenchmarksWithoutDataTransfer<int>(
+ (fun context -> singleSource context ArithmeticOperations.intSumOption ArithmeticOperations.intMulOption),
+ int32,
+ (fun _ -> Utils.nextInt (System.Random())),
+ 0)
+
+ static member InputMatrixProvider =
+ BFSBenchmarks<_>.InputMatrixProviderBuilder "BFSBenchmarks.txt"
+
+type BFSIntWithTransferBenchmark() =
+
+ inherit BFSBenchmarksWithTransfer<int>(
+ (fun context -> singleSource context ArithmeticOperations.intSumOption ArithmeticOperations.intMulOption),
+ int32,
+ (fun _ -> Utils.nextInt (System.Random())),
+ 0)
+
+ static member InputMatrixProvider =
+ BFSBenchmarks<_>.InputMatrixProviderBuilder "BFSBenchmarks.txt"