package main

import "testing"

func testGeneratePrimes(upperLimit int64, useTensors bool, b *testing.B) {
	var config Config
	config.DontLoad = true
	config.UseTensors = useTensors

	for i := 0; i < b.N; i++ {
		generatePrimes(upperLimit, config)
	}
}

func BenchmarkPrimes1000CPU(b *testing.B) {
	testGeneratePrimes(1000, false, b)
}

func BenchmarkPrimes1000GPU(b *testing.B) {
	testGeneratePrimes(1000, true, b)
}