Skip to content
Snippets Groups Projects
Commit d5f19eae authored by Martin Lange's avatar Martin Lange
Browse files

run benchmarks only be file prefix

parent a724e7b3
No related branches found
No related tags found
1 merge request!237Optimize units etc.
import datetime as dt
import numpy as np
import finam as fm
if __name__ == "__main__":
start_time = dt.datetime(2000, 1, 1)
end_time = dt.datetime(2002, 1, 1)
size = (1024, 1024)
info1 = fm.Info(time=None, grid=fm.UniformGrid(size), units="m")
data = fm.data.to_xarray(fm.data.full(0.0, info1), info1)
def gen_data(t):
return np.copy(data)
source = fm.modules.CallbackGenerator(
callbacks={"Out": (gen_data, info1.copy())},
start=start_time,
step=dt.timedelta(days=1),
)
sink = fm.modules.DebugConsumer(
inputs={
"In": info1.copy(),
},
start=start_time,
step=dt.timedelta(days=365),
)
composition = fm.Composition([source, sink])
composition.initialize()
source["Out"] >> sink["In"]
composition.run(end_time=end_time)
......@@ -6,11 +6,14 @@ mkdir -p prof
for filename in benchmarks/profiling/*.py; do
fn=$(basename -- "$filename")
fn="${fn%.*}"
echo "$fn"
python benchmarks/profiling/"$fn".py prof/"$fn".pstats
gprof2dot --colour-nodes-by-selftime -f pstats prof/"$fn".pstats > prof/"$fn".dot
dot -Tsvg -o prof/"$fn".svg prof/"$fn".dot
dot -Tpng -o prof/"$fn".png prof/"$fn".dot
if [[ "$fn" == profile_* ]] ;
then
echo "$fn"
python benchmarks/profiling/"$fn".py prof/"$fn".pstats
gprof2dot --colour-nodes-by-selftime -f pstats prof/"$fn".pstats > prof/"$fn".dot
dot -Tsvg -o prof/"$fn".svg prof/"$fn".dot
dot -Tpng -o prof/"$fn".png prof/"$fn".dot
fi
done
python benchmarks/pstats_to_csv.py
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment