Skip to content

Commit 573e145

Browse files
committed
document changes to make
Signed-off-by: Michael Kalantar <kalantar@us.ibm.com>
1 parent 339289b commit 573e145

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

setup/functions.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1326,7 +1326,8 @@ def add_affinity(ev:dict, section_indent: str = "") -> str:
13261326
identifier = "common"
13271327

13281328
if ev["control_environment_type_modelservice_active"]:
1329-
1329+
# use LLMDBENCH_VLLM_COMMON_AFFINITY to
1330+
# create acceleratorTypes: {labelKey: , labelValues []}
13301331
affinity = ev["vllm_common_affinity"]
13311332
if ":" in affinity:
13321333
affinity_key, affinity_value = affinity.split(":", 1)
@@ -1344,6 +1345,12 @@ def add_affinity(ev:dict, section_indent: str = "") -> str:
13441345
else :
13451346
return f"{section_indent}accelerator:\n{section_indent} type: {acellerator_type}\n{section_indent} resources:\n{section_indent} {acellerator_type}: \"{acellerator_product}\""
13461347

1348+
def add_accelerator():
1349+
# take LLMDBENCH_VLLM_COMMON_ACCELERATOR_RESOURCE and create
1350+
# accelerator: { type: ..., resources: {} }
1351+
pass
1352+
1353+
13471354
def add_additional_env_to_yaml(ev: dict, env_vars_string: str) -> str:
13481355
"""
13491356
Generate additional environment variables YAML.

0 commit comments

Comments
 (0)