File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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+
13471354def add_additional_env_to_yaml (ev : dict , env_vars_string : str ) -> str :
13481355 """
13491356 Generate additional environment variables YAML.
You can’t perform that action at this time.
0 commit comments