// Copyright (c) 2016- University of Oxford (Atilim Gunes Baydin <gunes@robots.ox.ac.uk>)

// and other contributors, see LICENSE in root of repository.

//

// BSD 2-Clause License. See LICENSE in root of repository.

open System.IO

open BenchmarkDotNet.Running

open DiffSharp.Benchmarks.Python

[<EntryPoint>]

let main args =

let summaries = BenchmarkSwitcher.FromAssembly(System.Reflection.Assembly.GetExecutingAssembly()).Run(args)

let lines =

[ for summary in summaries do

for case in summary.BenchmarksCases do

let v =

try

if case.Descriptor <> null &&

case.Descriptor.Categories <> null &&

case.Descriptor.Categories.Length > 0 then

if summary <> null && (try (summary[case] |> ignore); true with _ -> false) then

let report = summary[case]

let tensorSize = case.Parameters["tensorSize"] :?> int

let dtypeName = case.Parameters["dtypeName"] :?> string

let deviceName = case.Parameters["deviceName"] :?> string

// get the time in milliseconds

let runtime = report.ResultStatistics.Mean / 1000000.0 |> int64

let nm = case.Descriptor.Categories[0]

let key = nm + string tensorSize + dtypeName + deviceName

Some (sprintf "%s,%d" key runtime)

else

None

else

None

with _ -> None

match v with

| None -> ()

| Some r -> yield r

]

File.WriteAllLines(Path.Combine(__SOURCE_DIRECTORY__, "results.csv"), lines)

0