1 // Copyright 2012 The Go Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style 3 // license that can be found in the LICENSE file. 4 5 // This file implements a GCD benchmark. 6 // Usage: go test math/big -test.bench GCD 7 8 package big 9 10 import ( 11 "math/rand" 12 "testing" 13 ) 14 15 // randInt returns a pseudo-random Int in the range [1<<(size-1), (1<<size) - 1] 16 func randInt(r *rand.Rand, size uint) *Int { 17 n := new(Int).Lsh(intOne, size-1) 18 x := new(Int).Rand(r, n) 19 return x.Add(x, n) // make sure result > 1<<(size-1) 20 } 21 22 func runGCD(b *testing.B, aSize, bSize uint) { 23 b.StopTimer() 24 var r = rand.New(rand.NewSource(1234)) 25 aa := randInt(r, aSize) 26 bb := randInt(r, bSize) 27 b.StartTimer() 28 for i := 0; i < b.N; i++ { 29 new(Int).GCD(nil, nil, aa, bb) 30 } 31 } 32 33 func BenchmarkGCD10x10(b *testing.B) { runGCD(b, 10, 10) } 34 func BenchmarkGCD10x100(b *testing.B) { runGCD(b, 10, 100) } 35 func BenchmarkGCD10x1000(b *testing.B) { runGCD(b, 10, 1000) } 36 func BenchmarkGCD10x10000(b *testing.B) { runGCD(b, 10, 10000) } 37 func BenchmarkGCD10x100000(b *testing.B) { runGCD(b, 10, 100000) } 38 func BenchmarkGCD100x100(b *testing.B) { runGCD(b, 100, 100) } 39 func BenchmarkGCD100x1000(b *testing.B) { runGCD(b, 100, 1000) } 40 func BenchmarkGCD100x10000(b *testing.B) { runGCD(b, 100, 10000) } 41 func BenchmarkGCD100x100000(b *testing.B) { runGCD(b, 100, 100000) } 42 func BenchmarkGCD1000x1000(b *testing.B) { runGCD(b, 1000, 1000) } 43 func BenchmarkGCD1000x10000(b *testing.B) { runGCD(b, 1000, 10000) } 44 func BenchmarkGCD1000x100000(b *testing.B) { runGCD(b, 1000, 100000) } 45 func BenchmarkGCD10000x10000(b *testing.B) { runGCD(b, 10000, 10000) } 46 func BenchmarkGCD10000x100000(b *testing.B) { runGCD(b, 10000, 100000) } 47 func BenchmarkGCD100000x100000(b *testing.B) { runGCD(b, 100000, 100000) } 48
