Skip to content

Commit fc41656

Browse files
committed
add a prefixing to benchmarks
1 parent 2e30981 commit fc41656

File tree

1 file changed

+23
-2
lines changed

1 file changed

+23
-2
lines changed

benchmark/make_epiaware_suite.jl

Lines changed: 23 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,28 @@
33
"""
44
function make_epiaware_suite(model; check = true,
55
adbackends = [:forwarddiff, :reversediff, :reversediff_compiled])
6-
suite = TuringBenchmarking.make_turing_suite(
7-
model; check = check, adbackends = adbackends)
6+
suite = prefix_warnings(
7+
() -> TuringBenchmarking.make_turing_suite(
8+
model; check = check, adbackends = adbackends),
9+
model
10+
)
811
return suite
912
end
13+
14+
function prefix_warnings(f, g)
15+
original_stderr = stderr
16+
(read_pipe, write_pipe) = redirect_stderr()
17+
result = nothing
18+
try
19+
result = f()
20+
finally
21+
redirect_stderr(original_stderr)
22+
close(write_pipe)
23+
output = String(read(read_pipe))
24+
if !isempty(output)
25+
println(stderr, "\nWarnings from $(g):")
26+
println(stderr, output)
27+
end
28+
end
29+
return result
30+
end

0 commit comments

Comments
 (0)