diff options
Diffstat (limited to 'test/bench/spectral-norm.go')
| -rw-r--r-- | test/bench/spectral-norm.go | 38 |
1 files changed, 19 insertions, 19 deletions
diff --git a/test/bench/spectral-norm.go b/test/bench/spectral-norm.go index e79bc282e..6667f3e04 100644 --- a/test/bench/spectral-norm.go +++ b/test/bench/spectral-norm.go @@ -37,20 +37,20 @@ POSSIBILITY OF SUCH DAMAGE. package main import ( - "flag"; - "fmt"; - "math"; + "flag" + "fmt" + "math" ) var n = flag.Int("n", 2000, "count") -func evalA(i, j int) float64 { return 1 / float64(((i+j)*(i+j+1)/2 + i + 1)) } +func evalA(i, j int) float64 { return 1 / float64(((i+j)*(i+j+1)/2 + i + 1)) } type Vec []float64 func (v Vec) Times(u Vec) { for i := 0; i < len(v); i++ { - v[i] = 0; + v[i] = 0 for j := 0; j < len(u); j++ { v[i] += evalA(i, j) * u[j] } @@ -59,7 +59,7 @@ func (v Vec) Times(u Vec) { func (v Vec) TimesTransp(u Vec) { for i := 0; i < len(v); i++ { - v[i] = 0; + v[i] = 0 for j := 0; j < len(u); j++ { v[i] += evalA(j, i) * u[j] } @@ -67,27 +67,27 @@ func (v Vec) TimesTransp(u Vec) { } func (v Vec) ATimesTransp(u Vec) { - x := make(Vec, len(u)); - x.Times(u); - v.TimesTransp(x); + x := make(Vec, len(u)) + x.Times(u) + v.TimesTransp(x) } func main() { - flag.Parse(); - N := *n; - u := make(Vec, N); + flag.Parse() + N := *n + u := make(Vec, N) for i := 0; i < N; i++ { u[i] = 1 } - v := make(Vec, N); + v := make(Vec, N) for i := 0; i < 10; i++ { - v.ATimesTransp(u); - u.ATimesTransp(v); + v.ATimesTransp(u) + u.ATimesTransp(v) } - var vBv, vv float64; + var vBv, vv float64 for i := 0; i < N; i++ { - vBv += u[i] * v[i]; - vv += v[i] * v[i]; + vBv += u[i] * v[i] + vv += v[i] * v[i] } - fmt.Printf("%0.9f\n", math.Sqrt(vBv/vv)); + fmt.Printf("%0.9f\n", math.Sqrt(vBv/vv)) } |
