luau/bench/tests/sieve.lua
Vyacheslav Egorov ce1800746b
Fix 'require' in benchmarks to work with new relative system (#1120)
Old style require is now called with `pcall` to support comparing
against Lua.
New style require is now a third option.

Edit: this will be a temporary solution until the 'paths' support in
.luaurc is fixed.
2023-12-02 14:20:54 -08:00

44 lines
No EOL
1.1 KiB
Lua

local bench = script and require(script.Parent.bench_support) or pcall(require, "bench_support") or require("../bench_support")
function test()
-- the sieve of of Eratosthenes programmed with coroutines
-- typical usage: lua -e N=1000 sieve.lua | column
-- generate all the numbers from 2 to n
function gen (n)
return coroutine.wrap(function ()
for i=2,n do coroutine.yield(i) end
end)
end
-- filter the numbers generated by `g', removing multiples of `p'
function filter (p, g)
return coroutine.wrap(function ()
while 1 do
local n = g()
if n == nil then return end
if n % p ~= 0 then coroutine.yield(n) end
end
end)
end
local ts0 = os.clock()
for loops=1,100 do
N = 1000
x = gen(N) -- generate primes up to N
while 1 do
local n = x() -- pick a number until done
if n == nil then break end
-- print(n) -- must be a prime number
x = filter(n, x) -- now remove its multiples
end
end
local ts1 = os.clock()
return ts1-ts0
end
bench.runCode(test, "sieve")