Skip to content

Commit 5ddb405

Browse files
committed
Reduce runtime of RooFit benchmarks
1 parent fb1fc3a commit 5ddb405

File tree

2 files changed

+11
-1
lines changed

2 files changed

+11
-1
lines changed

root/roofit/roofit/benchCodeSquashAD.cxx

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -155,12 +155,22 @@ static void BM_RooFuncWrapper_ManyParams_Minimization(benchmark::State &state)
155155

156156
int main(int argc, char **argv)
157157
{
158+
// If you want to make a more complete scaling study, you can scale up to a larger number of channels:
159+
// #define LARGE_STUDY
160+
161+
#ifdef LARGE_STUDY
158162
std::vector<long int> rangeLow{1, 2, 3, 4, 5, 6, 7, 8, 12, 17, 23, 28, 33, 38, 43, 49, 54, 59};
163+
#else
164+
std::vector<long int> rangeLow{1, 3, 5, 7, 12, 23, 33, 43, 54};
165+
#endif
159166
std::vector<long int> rangeHigh{64, 70, 75, 80, 85, 90, 96, 100, 110, 120, 130, 140,
160167
150, 160, 170, 180, 190, 200, 225, 250, 275, 300, 325};
161168
std::vector<long int> range;
162169
range.insert(range.end(), rangeLow.begin(), rangeLow.end());
170+
171+
#ifdef LARGE_STUDY
163172
range.insert(range.end(), rangeHigh.begin(), rangeHigh.end());
173+
#endif
164174

165175
// Run the minimization with the reference NLL
166176
RooFitADBenchmarksUtils::doBenchmarks("NllReferenceMinimization", Reference,

root/roofit/roofit/benchRooFitBackends.cxx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
#include <benchmark/benchmark.h>
2323

2424
int printLevel = 0;
25-
size_t nEvents = 100000;
25+
size_t nEvents = 10000;
2626
const auto minimizerName = "Minuit2";
2727

2828
enum RunConfig_t { runScalar, runCpu, fitScalar, fitCpu, fitCuda };

0 commit comments

Comments
 (0)