|
1 | 1 | import 'package:collection/collection.dart';
|
2 | 2 |
|
3 |
| -import 'package:mlperfbench/app_constants.dart'; |
4 | 3 | import 'package:mlperfbench/backend/bridge/run_settings.dart';
|
5 | 4 | import 'package:mlperfbench/backend/loadgen_info.dart';
|
6 | 5 | import 'package:mlperfbench/benchmark/info.dart';
|
@@ -69,24 +68,13 @@ class Benchmark {
|
69 | 68 | required List<pb.CommonSetting> commonSettings,
|
70 | 69 | required String backendLibName,
|
71 | 70 | required String logDir,
|
72 |
| - required int testMinDuration, |
73 |
| - required int testMinQueryCount, |
74 | 71 | }) async {
|
75 | 72 | final dataset = runMode.chooseDataset(taskConfig);
|
| 73 | + final runConfig = runMode.chooseRunConfig(taskConfig); |
76 | 74 |
|
77 |
| - int minQueryCount; |
78 |
| - double minDuration; |
79 |
| - if (testMinDuration != 0) { |
80 |
| - minQueryCount = testMinQueryCount; |
81 |
| - minDuration = testMinDuration.toDouble(); |
82 |
| - } else if (DartDefine.isFastMode) { |
83 |
| - minQueryCount = 8; |
84 |
| - minDuration = 1.0; |
85 |
| - } else { |
86 |
| - minQueryCount = taskConfig.minQueryCount; |
87 |
| - minDuration = taskConfig.minDuration; |
88 |
| - } |
89 |
| - double maxDuration = taskConfig.maxDuration; |
| 75 | + int minQueryCount = runConfig.minQueryCount; |
| 76 | + double minDuration = runConfig.minDuration; |
| 77 | + double maxDuration = runConfig.maxDuration; |
90 | 78 |
|
91 | 79 | final settings = pb.SettingList(
|
92 | 80 | setting: commonSettings,
|
@@ -114,7 +102,7 @@ class Benchmark {
|
114 | 102 | model_image_width: taskConfig.model.imageWidth,
|
115 | 103 | model_image_height: taskConfig.model.imageHeight,
|
116 | 104 | scenario: taskConfig.scenario,
|
117 |
| - mode: runMode.loadgenMode, |
| 105 | + mode: runMode.loadgenMode.name, |
118 | 106 | batch_size: selectedDelegate.batchSize,
|
119 | 107 | min_query_count: minQueryCount,
|
120 | 108 | min_duration: minDuration,
|
|
0 commit comments