Fix perf analyzer config and tests

debermudez · debermudez · commit f60bf049112f · 2025-03-14T17:02:50.000-07:00
diff --git a/genai-perf/genai_perf/config/generate/perf_analyzer_config.py b/genai-perf/genai_perf/config/generate/perf_analyzer_config.py
@@ -52,6 +52,7 @@
     "batch_size_audio",
     "batch_size_image",
     "batch_size_text",
+    "block_size",
     "concurrency",
     "endpoint_type",
     "extra_inputs",
diff --git a/genai-perf/tests/test_exporters/test_json_exporter.py b/genai-perf/tests/test_exporters/test_json_exporter.py
@@ -212,6 +212,7 @@ def test_generate_json_input_config(
             "subcommand": "profile",
             "prompt_source": "synthetic",
             "extra_inputs": {},
+            "block_size": 512,
         }
 
         _, data = next(iter(mock_read_write))
diff --git a/genai-perf/tests/test_retrievers/test_payload_input_retriever.py b/genai-perf/tests/test_retrievers/test_payload_input_retriever.py
@@ -53,6 +53,7 @@ def __init__(self):
                 self.payload_input_filename = Path("test_input.jsonl")
                 self.prompt_tokens_mean = 10
                 self.prompt_tokens_stddev = 2
+                self.block_size = 10
 
         return MockConfig()
 

Original file line number	Diff line number	Diff line change
`@@ -212,6 +212,7 @@ def test_generate_json_input_config(`
`212`	`212`	`"subcommand": "profile",`
`213`	`213`	`"prompt_source": "synthetic",`
`214`	`214`	`"extra_inputs": {},`
	`215`	`+ "block_size": 512,`
`215`	`216`	`}`
`216`	`217`
`217`	`218`	`_, data = next(iter(mock_read_write))`