File tree Expand file tree Collapse file tree 2 files changed +6
-4
lines changed
src/vivarium_profiling/tools Expand file tree Collapse file tree 2 files changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -79,7 +79,6 @@ def run_single_benchmark(
7979 run_number : int ,
8080 total_runs : int ,
8181 spec_results_dir : str ,
82- model_spec_name : str ,
8382 config : ExtractionConfig ,
8483) -> dict [str , Any ]:
8584 """Run a single benchmark iteration.
@@ -94,8 +93,6 @@ def run_single_benchmark(
9493 Total number of runs for this spec.
9594 spec_results_dir
9695 Directory to store results for this spec.
97- model_spec_name
98- Name of the model spec (stem of the file).
9996 config
10097 Extraction configuration. Defaults to DEFAULT_CONFIG.
10198
@@ -122,6 +119,7 @@ def run_single_benchmark(
122119 move_mprof_files (current_results_dir )
123120
124121 # Get runtime and function metrics
122+ model_spec_name = Path (spec ).stem
125123 stats_file = Path (current_results_dir ) / f"{ model_spec_name } .stats"
126124 stats_file_txt = f"{ stats_file } .txt"
127125
@@ -211,7 +209,6 @@ def run_benchmark_loop(
211209 run ,
212210 num_runs ,
213211 str (spec_specific_results_dir ),
214- model_spec_name ,
215212 config ,
216213 )
217214 result_df = pd .DataFrame ([results ])
Original file line number Diff line number Diff line change 2525 "population_get_cumtime" ,
2626 "population_get_percall" ,
2727 "population_get_ncalls" ,
28+ "rt_setup_s" ,
29+ "rt_initialize_simulants_s" ,
30+ "rt_run_s" ,
31+ "rt_finalize_s" ,
32+ "rt_report_s" ,
2833]
2934
3035
You can’t perform that action at this time.
0 commit comments