Skip to content

Commit

Permalink
Profiling info processor: allow to specify non-default params
Browse files Browse the repository at this point in the history
  • Loading branch information
m-kus committed Oct 6, 2024
1 parent 3900e0c commit 174b764
Show file tree
Hide file tree
Showing 4 changed files with 10 additions and 8 deletions.
1 change: 1 addition & 0 deletions crates/bin/cairo-run/src/main.rs
Original file line number Diff line number Diff line change
Expand Up @@ -101,6 +101,7 @@ fn main() -> anyhow::Result<()> {
Some(db),
sierra_program,
debug_info.statements_locations.get_statements_functions_map_for_tests(db),
Default::default(),
);
match result.profiling_info {
Some(raw_profiling_info) => {
Expand Down
8 changes: 2 additions & 6 deletions crates/cairo-lang-runner/src/profiling.rs
Original file line number Diff line number Diff line change
Expand Up @@ -237,13 +237,9 @@ impl<'a> ProfilingInfoProcessor<'a> {
db: Option<&'a dyn SierraGenGroup>,
sierra_program: Program,
statements_functions: UnorderedHashMap<StatementIdx, String>,
params: ProfilingInfoProcessorParams,
) -> Self {
Self {
db,
sierra_program,
statements_functions,
params: ProfilingInfoProcessorParams::default(),
}
Self { db, sierra_program, statements_functions, params }
}

/// Processes the raw profiling info according to the params set in the processor.
Expand Down
8 changes: 6 additions & 2 deletions crates/cairo-lang-runner/src/profiling_test.rs
Original file line number Diff line number Diff line change
Expand Up @@ -68,8 +68,12 @@ pub fn test_profiling(
let result = runner
.run_function_with_starknet_context(func, &[], Some(u32::MAX as usize), Default::default())
.unwrap();
let profiling_processor =
ProfilingInfoProcessor::new(Some(&db), sierra_program, statements_functions);
let profiling_processor = ProfilingInfoProcessor::new(
Some(&db),
sierra_program,
statements_functions,
Default::default(),
);
let processed_profiling_info = profiling_processor.process(&result.profiling_info.unwrap());

TestRunnerResult {
Expand Down
1 change: 1 addition & 0 deletions crates/cairo-lang-test-runner/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -544,6 +544,7 @@ fn update_summary(
Some(*db),
sierra_program.clone(),
statements_functions.clone(),
Default::default(),
);
let processed_profiling_info =
profiling_processor.process_ex(&profiling_info, profiling_params);
Expand Down

0 comments on commit 174b764

Please sign in to comment.