+++ /dev/null
-// Copyright 2014 The Gonum Authors. All rights reserved.
-// Use of this code is governed by a BSD-style
-// license that can be found in the LICENSE file
-
-// Script for automatic code generation of the benchmark routines.
-package main
-
-import (
- "fmt"
- "os"
- "os/exec"
- "path"
- "path/filepath"
- "strconv"
-)
-
-var gopath string
-
-var copyrightnotice = []byte(`// Copyright 2014 The Gonum Authors. All rights reserved.
-// Use of this code is governed by a BSD-style
-// license that can be found in the LICENSE file`)
-
-var autogen = []byte("// Code generated by \"go run gonum.org/v1/gonum/blas/testblas/benchautogen/autogen_bench_level1double.go\"; DO NOT EDIT.\n")
-
-var imports = []byte(`import(
- "golang.org/x/exp/rand"
- "testing"
-
- "gonum.org/v1/gonum/blas"
-)`)
-
-var randomSliceFunction = []byte(`func randomSlice(l, idx int) ([]float64) {
- if idx < 0{
- idx = -idx
- }
- s := make([]float64, l * idx)
- for i := range s {
- s[i] = rand.Float64()
- }
- return s
-}`)
-
-const (
- posInc1 = 5
- posInc2 = 3
- negInc1 = -3
- negInc2 = -4
-)
-
-var level1Sizes = []struct {
- lower string
- upper string
- camel string
- size int
-}{
- {
- lower: "small",
- upper: "SMALL_SLICE",
- camel: "Small",
- size: 10,
- },
- {
- lower: "medium",
- upper: "MEDIUM_SLICE",
- camel: "Medium",
- size: 1000,
- },
- {
- lower: "large",
- upper: "LARGE_SLICE",
- camel: "Large",
- size: 100000,
- },
- {
- lower: "huge",
- upper: "HUGE_SLICE",
- camel: "Huge",
- size: 10000000,
- },
-}
-
-type level1functionStruct struct {
- camel string
- sig string
- call string
- extraSetup string
- oneInput bool
- extraName string // if have a couple different cases for the same function
-}
-
-var level1Functions = []level1functionStruct{
- {
- camel: "Ddot",
- sig: "n int, x []float64, incX int, y []float64, incY int",
- call: "n, x, incX, y, incY",
- oneInput: false,
- },
- {
- camel: "Dnrm2",
- sig: "n int, x []float64, incX int",
- call: "n, x, incX",
- oneInput: true,
- },
- {
- camel: "Dasum",
- sig: "n int, x []float64, incX int",
- call: "n, x, incX",
- oneInput: true,
- },
- {
- camel: "Idamax",
- sig: "n int, x []float64, incX int",
- call: "n, x, incX",
- oneInput: true,
- },
- {
- camel: "Dswap",
- sig: "n int, x []float64, incX int, y []float64, incY int",
- call: "n, x, incX, y, incY",
- oneInput: false,
- },
- {
- camel: "Dcopy",
- sig: "n int, x []float64, incX int, y []float64, incY int",
- call: "n, x, incX, y, incY",
- oneInput: false,
- },
- {
- camel: "Daxpy",
- sig: "n int, alpha float64, x []float64, incX int, y []float64, incY int",
- call: "n, alpha, x, incX, y, incY",
- extraSetup: "alpha := 2.4",
- oneInput: false,
- },
- {
- camel: "Drot",
- sig: "n int, x []float64, incX int, y []float64, incY int, c, s float64",
- call: "n, x, incX, y, incY, c, s",
- extraSetup: "c := 0.89725836967\ns:= 0.44150585279",
- oneInput: false,
- },
- {
- camel: "Drotm",
- sig: "n int, x []float64, incX int, y []float64, incY int, p blas.DrotmParams",
- call: "n, x, incX, y, incY, p",
- extraSetup: "p := blas.DrotmParams{Flag: blas.OffDiagonal, H: [4]float64{0, -0.625, 0.9375,0}}",
- oneInput: false,
- extraName: "OffDia",
- },
- {
- camel: "Drotm",
- sig: "n int, x []float64, incX int, y []float64, incY int, p blas.DrotmParams",
- call: "n, x, incX, y, incY, p",
- extraSetup: "p := blas.DrotmParams{Flag: blas.OffDiagonal, H: [4]float64{5.0 / 12, 0, 0, 0.625}}",
- oneInput: false,
- extraName: "Dia",
- },
- {
- camel: "Drotm",
- sig: "n int, x []float64, incX int, y []float64, incY int, p blas.DrotmParams",
- call: "n, x, incX, y, incY, p",
- extraSetup: "p := blas.DrotmParams{Flag: blas.OffDiagonal, H: [4]float64{4096, -3584, 1792, 4096}}",
- oneInput: false,
- extraName: "Resc",
- },
- {
- camel: "Dscal",
- sig: "n int, alpha float64, x []float64, incX int",
- call: "n, alpha, x, incX",
- extraSetup: "alpha := 2.4",
- oneInput: true,
- },
-}
-
-func init() {
- gopath = os.Getenv("GOPATH")
- if gopath == "" {
- panic("gopath not set")
- }
-}
-
-func main() {
- blasPath := filepath.Join(gopath, "src", "gonum.org", "v1", "gonum", "blas")
-
- pkgs := []struct{ name string }{{name: "native"}, {name: "cgo"}}
-
- for _, pkg := range pkgs {
- err := level1(filepath.Join(blasPath, pkg.name), pkg.name)
- if err != nil {
- fmt.Println(err)
- os.Exit(1)
- }
-
- err = exec.Command("go", "fmt", path.Join("gonum.org", "v1", "gonum", "blas", pkg.name)).Run()
- if err != nil {
- fmt.Println(err)
- os.Exit(1)
- }
- }
-}
-
-func printHeader(f *os.File, name string) error {
- if _, err := f.Write(autogen); err != nil {
- return err
- }
- f.WriteString("\n\n")
- f.Write(copyrightnotice)
- f.WriteString("\n\n")
- f.WriteString("package " + name)
- f.WriteString("\n\n")
- f.Write(imports)
- f.WriteString("\n\n")
- return nil
-}
-
-// Generate the benchmark scripts for level1
-func level1(benchPath string, pkgname string) error {
- // Generate level 1 benchmarks
- level1Filepath := filepath.Join(benchPath, "level1doubleBench_auto_test.go")
- f, err := os.Create(level1Filepath)
- if err != nil {
- fmt.Println(err)
- os.Exit(1)
- }
- defer f.Close()
- printHeader(f, pkgname)
-
- // Print all of the constants
- f.WriteString("const (\n")
- f.WriteString("\tposInc1 = " + strconv.Itoa(posInc1) + "\n")
- f.WriteString("\tposInc2 = " + strconv.Itoa(posInc2) + "\n")
- f.WriteString("\tnegInc1 = " + strconv.Itoa(negInc1) + "\n")
- f.WriteString("\tnegInc2 = " + strconv.Itoa(negInc2) + "\n")
- for _, con := range level1Sizes {
- f.WriteString("\t" + con.upper + " = " + strconv.Itoa(con.size) + "\n")
- }
- f.WriteString(")\n")
- f.WriteString("\n")
-
- // Write the randomSlice function
- f.Write(randomSliceFunction)
- f.WriteString("\n\n")
-
- // Start writing the benchmarks
- for _, fun := range level1Functions {
- writeLevel1Benchmark(fun, f)
- f.WriteString("\n/* ------------------ */ \n")
- }
-
- return nil
-}
-
-func writeLevel1Benchmark(fun level1functionStruct, f *os.File) {
- // First, write the base benchmark file
- f.WriteString("func benchmark" + fun.camel + fun.extraName + "(b *testing.B, ")
- f.WriteString(fun.sig)
- f.WriteString(") {\n")
-
- f.WriteString("b.ResetTimer()\n")
- f.WriteString("for i := 0; i < b.N; i++{\n")
- f.WriteString("\timpl." + fun.camel + "(")
-
- f.WriteString(fun.call)
- f.WriteString(")\n}\n}\n")
- f.WriteString("\n")
-
- // Write all of the benchmarks to call it
- for _, sz := range level1Sizes {
- lambda := func(incX, incY, name string, twoInput bool) {
- f.WriteString("func Benchmark" + fun.camel + fun.extraName + sz.camel + name + "(b *testing.B){\n")
- f.WriteString("n := " + sz.upper + "\n")
- f.WriteString("incX := " + incX + "\n")
- f.WriteString("x := randomSlice(n, incX)\n")
- if twoInput {
- f.WriteString("incY := " + incY + "\n")
- f.WriteString("y := randomSlice(n, incY)\n")
- }
- f.WriteString(fun.extraSetup + "\n")
- f.WriteString("benchmark" + fun.camel + fun.extraName + "(b, " + fun.call + ")\n")
- f.WriteString("}\n\n")
- }
- if fun.oneInput {
- lambda("1", "", "UnitaryInc", false)
- lambda("posInc1", "", "PosInc", false)
- } else {
- lambda("1", "1", "BothUnitary", true)
- lambda("posInc1", "1", "IncUni", true)
- lambda("1", "negInc1", "UniInc", true)
- lambda("posInc1", "negInc1", "BothInc", true)
- }
- }
-}
OSDN Git Service
