Update test_dist_inference.py

microsoft · Dec 14, 2023 · 6b9fdf2 · 6b9fdf2
1 parent 06e65ad
commit 6b9fdf2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/benchmarks/micro_benchmarks/test_dist_inference.py b/tests/benchmarks/micro_benchmarks/test_dist_inference.py
@@ -193,7 +193,7 @@ def _test_dist_inference_result_parsing(self, platform, test_raw_output):
         (benchmark_class,
          predefine_params) = BenchmarkRegistry._BenchmarkRegistry__select_benchmark(benchmark_name, platform)
         assert (benchmark_class)
-        benchmark = benchmark_class(benchmark_name, parameters='')
+        benchmark = benchmark_class(benchmark_name, parameters='--num_layers=100')
         assert (benchmark)
         ret = benchmark._preprocess()
         assert (ret is True)