From edbceab57b66c5b442e7cae5212e351770f6b9d3 Mon Sep 17 00:00:00 2001 From: braf Date: Tue, 22 Aug 2023 21:39:45 +0000 Subject: [PATCH] Fixing L0 test --- qa/L0_config_search/test_config_generator.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/qa/L0_config_search/test_config_generator.py b/qa/L0_config_search/test_config_generator.py index 5ca2d689c..6fa72f912 100755 --- a/qa/L0_config_search/test_config_generator.py +++ b/qa/L0_config_search/test_config_generator.py @@ -70,7 +70,7 @@ def generate_search_disable(self): total_param_count = self._calculate_total_params( concurrency_count, instance_count ) - self._write_file(total_param_count, 1, 2, 1, model_config) + self._write_file(total_param_count, total_param_count, 2, 2, model_config) def generate_max_limit_with_model_config(self): concurrency_count = 2 @@ -91,7 +91,7 @@ def generate_max_limit_with_model_config(self): total_param_count = self._calculate_total_params( concurrency_count, instance_count ) - self._write_file(total_param_count, 2, 2, 1, model_config) + self._write_file(total_param_count, total_param_count, 2, 2, model_config) def generate_max_limit(self): concurrency_count = 2 @@ -105,7 +105,7 @@ def generate_max_limit(self): total_param_count = self._calculate_total_params( concurrency_count, instance_count ) - self._write_file(total_param_count, 2, 8, 1, model_config) + self._write_file(total_param_count, total_param_count, 8, 8, model_config) def generate_max_limit_with_param(self): concurrency_count = 1 # 1 because concurrency parameter is 1 entry below @@ -124,7 +124,7 @@ def generate_max_limit_with_param(self): total_param_count = self._calculate_total_params( concurrency_count, instance_count ) - self._write_file(total_param_count, 1, 6, 1, model_config) + self._write_file(total_param_count, total_param_count, 6, 6, model_config) def generate_max_limit_with_param_and_model_config(self): concurrency_count = 1 # 1 because concurrency parameter is 1 entry below @@ -146,7 +146,7 @@ def generate_max_limit_with_param_and_model_config(self): total_param_count = self._calculate_total_params( concurrency_count, instance_count ) - self._write_file(total_param_count, 1, 2, 1, model_config) + self._write_file(total_param_count, total_param_count, 2, 2, model_config) def generate_max_limit_with_dynamic_batch_disable(self): concurrency_count = 2 @@ -160,7 +160,7 @@ def generate_max_limit_with_dynamic_batch_disable(self): total_param_count = self._calculate_total_params( concurrency_count, instance_count ) - self._write_file(total_param_count, 2, 4, 1, model_config) + self._write_file(total_param_count, total_param_count, 4, 4, model_config) def _calculate_total_params( self, concurrency_count, instance_count, default_config_count=1