Skip to content

Commit 936145b

Browse files
bo-nvdominicshanshan
authored andcommitted
[None][fix] Update tests to use standardized uppercase backend identifiers (NVIDIA#6921)
Signed-off-by: Bo Deng <[email protected]>
1 parent 430a2ab commit 936145b

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

tests/integration/defs/accuracy/test_disaggregated_serving.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -627,13 +627,13 @@ def test_nixl_backend(self):
627627
ctx_server_config = {
628628
"disable_overlap_scheduler": True,
629629
"cache_transceiver_config": {
630-
"backend": "nixl"
630+
"backend": "NIXL"
631631
}
632632
}
633633
gen_server_config = {
634634
"disable_overlap_scheduler": True,
635635
"cache_transceiver_config": {
636-
"backend": "nixl"
636+
"backend": "NIXL"
637637
}
638638
}
639639
disaggregated_server_config = {
@@ -763,13 +763,13 @@ def test_nixl_backend(self):
763763
ctx_server_config = {
764764
"disable_overlap_scheduler": True,
765765
"cache_transceiver_config": {
766-
"backend": "nixl"
766+
"backend": "NIXL"
767767
}
768768
}
769769
gen_server_config = {
770770
"disable_overlap_scheduler": True,
771771
"cache_transceiver_config": {
772-
"backend": "nixl"
772+
"backend": "NIXL"
773773
}
774774
}
775775
disaggregated_server_config = {

tests/integration/defs/disaggregated/test_disaggregated.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1245,8 +1245,8 @@ def get_config_for_benchmark(model_root, backend):
12451245
def test_disaggregated_benchmark_on_diff_backends(
12461246
disaggregated_test_root, disaggregated_example_root, llm_venv,
12471247
benchmark_model_root, benchmark_root, shared_gpt_path):
1248-
nixl_config = get_config_for_benchmark(benchmark_model_root, "nixl")
1249-
ucx_config = get_config_for_benchmark(benchmark_model_root, "ucx")
1248+
nixl_config = get_config_for_benchmark(benchmark_model_root, "NIXL")
1249+
ucx_config = get_config_for_benchmark(benchmark_model_root, "UCX")
12501250
temp_dir = tempfile.TemporaryDirectory()
12511251
nixl_config_path = os.path.join(temp_dir.name, "nixl_config.yaml")
12521252
ucx_config_path = os.path.join(temp_dir.name, "ucx_config.yaml")

0 commit comments

Comments
 (0)