Skip to content

Latest commit

 

History

History
120 lines (104 loc) · 18.4 KB

File metadata and controls

120 lines (104 loc) · 18.4 KB

Concept

llm-d-benchmark provides its own automated framework for the standup of stacks serving large language models in a Kubernetes cluster.

Motivation

In order to allow reproducible and flexible experiments, and taking into account that the configuration paramaters have significant impact on the overall performance, it is necessary to provide the user with the ability to standup and teardown stacks.

Methods

Currently, two main standup methods are supported a) "Standalone", with multiple VLLM pods controlled by a deployment behind a single service b) "llm-d", which leverages a combination of llm-d-infra and llm-d-modelservice to deploy a full-fledged llm-d stack

Scenarios

All the information required for the standup of a stack is contained on a "scenario file". This information is encoded in the form of environment variables, with default values defined in config/defaults.yaml which can be then overriden inside a scenario file (YAML-based) or via specification templates (Jinja2 .yaml.j2 files).

Multiple steps

The full standup of a stack is a multi-step process. The lifecycle document go into more details explaning the meaning of each different individual step.

Use

A scenario file has to be manually crafted as a YAML file. Once crafted, it can be used by llmdbenchmark standup, llmdbenchmark run or llmdbenchmark teardown commands. Its access is controlled by the following parameters.

Note

llmdbenchmark experiment is a command that combines llmdbenchmark standup, llmdbenchmark run and llmdbenchmark teardown into a single operation. Therefore, the command line parameters supported by the former is a combination of the latter three.

The scenario parameters can be roughly categorized in four groups:

  • Target-specific (Cluster API access, authentication tokens, standup methods and models)
Variable Meaning Note
LLMDBENCH_CLUSTER_URL URL to API access to Kubernetes cluster "auto" means "current" (e.g. ~/.kube/config) is used
LLMDBENCH_CLUSTER_TOKEN Used to authenticate to the cluster Ignored for LLMDBENCH_CLUSTER_URL="auto"
LLMDBENCH_HF_TOKEN Hugging face token Required for gated models; optional for public models (auto-detected)
LLMDBENCH_DEPLOY_SCENARIO File containing multiple environment variables which will override defaults If not specified, defaults to (empty) none.yaml. Can be overriden with CLI parameter -c/--scenario
LLMDBENCH_DEPLOY_MODEL_LIST List (comma-separated values) of models to be run against Default=meta-llama/Llama-3.2-1B-Instruct. Can be overriden with CLI parameter -m/--models
LLMDBENCH_DEPLOY_METHODS List (comma-separated values) of standup methods Default=modelservice. Can be overriden with CLI parameter -t/--methods

Tip

In case the full path is ommited for the scenario file (either by setting LLMDBENCH_DEPLOY_SCENARIO or CLI parameter -c/--scenario, it is assumed that the scenario exists inside the config/scenarios folder

  • "Common" VLLM parameters, applicable to any standup method
Variable Meaning Note
LLMDBENCH_VLLM_COMMON_NAMESPACE Namespace where stack gets stood up Default=llmdbench. Can be overriden with CLI parameter -p/--namespace
LLMDBENCH_IGNORE_FAILED_VALIDATION Ignore failed sanity checks and continue to deployment Default=True. Capacity Planner will perform a sanity check on vLLM parameters such as valid TP, max-model-len, KV cache availability.
LLMDBENCH_VLLM_COMMON_ACCELERATOR_MEMORY GPU memory for LLMDBENCH_VLLM_COMMON_ACCELERATOR_RESOURCE (e.g. 80) Default=auto, will try to guess GPU memory from LLMDBENCH_VLLM_COMMON_ACCELERATOR_RESOURCE
LLMDBENCH_VLLM_COMMON_SERVICE_ACCOUNT Service Account for stack
LLMDBENCH_VLLM_COMMON_ACCELERATOR_RESOURCE Accelerator type (e.g., nvidia.com/gpu) "auto" means, will query the cluster to discover
LLMDBENCH_VLLM_COMMON_NETWORK_RESOURCE Network type (e.g., rdma/roce_gdr)
LLMDBENCH_VLLM_COMMON_VLLM_ALLOW_LONG_MAX_MODEL_LEN
LLMDBENCH_VLLM_COMMON_VLLM_SERVER_DEV_MODE e.g., 0, 1
LLMDBENCH_VLLM_COMMON_VLLM_LOAD_FORMAT e.g., safetensors, tensorizer, runai_streamer, fastsafetensors
LLMDBENCH_VLLM_COMMON_VLLM_LOGGING_LEVEL e.g., DEBUG, INFO, WARNING
LLMDBENCH_VLLM_COMMON_ENABLE_SLEEP_MODE e.g., true, false
LLMDBENCH_VLLM_COMMON_NETWORK_NR
LLMDBENCH_VLLM_COMMON_AFFINITY
LLMDBENCH_VLLM_COMMON_REPLICAS
LLMDBENCH_VLLM_COMMON_TENSOR_PARALLELISM
LLMDBENCH_VLLM_COMMON_DATA_PARALLELISM
LLMDBENCH_VLLM_COMMON_ACCELERATOR_NR
LLMDBENCH_VLLM_COMMON_ACCELERATOR_MEM_UTIL
LLMDBENCH_VLLM_COMMON_CPU_NR
LLMDBENCH_VLLM_COMMON_CPU_MEM
LLMDBENCH_VLLM_COMMON_MAX_MODEL_LEN
LLMDBENCH_VLLM_COMMON_BLOCK_SIZE
LLMDBENCH_VLLM_COMMON_MAX_NUM_BATCHED_TOKENS
LLMDBENCH_VLLM_COMMON_PVC_NAME
LLMDBENCH_VLLM_COMMON_PVC_STORAGE_CLASS
LLMDBENCH_VLLM_COMMON_PVC_MODEL_CACHE_SIZE
LLMDBENCH_VLLM_COMMON_PVC_DOWNLOAD_TIMEOUT
LLMDBENCH_VLLM_COMMON_HF_TOKEN_KEY
LLMDBENCH_VLLM_COMMON_HF_TOKEN_NAME
LLMDBENCH_VLLM_COMMON_INFERENCE_PORT
LLMDBENCH_VLLM_COMMON_FQDN
LLMDBENCH_VLLM_COMMON_TIMEOUT
LLMDBENCH_VLLM_COMMON_ANNOTATIONS
LLMDBENCH_VLLM_COMMON_ENVVARS_TO_YAML
LLMDBENCH_VLLM_COMMON_INITIAL_DELAY_PROBE
LLMDBENCH_VLLM_COMMON_POD_SCHEDULER
  • "Standalone"-specific VLLM parameters
Variable Meaning Note
LLMDBENCH_VLLM_COMMON_MODEL_LOADER_EXTRA_CONFIG
LLMDBENCH_VLLM_STANDALONE_PVC_MOUNTPOINT
LLMDBENCH_VLLM_STANDALONE_PREPROCESS e.g., source /setup/preprocess/standalone-preprocess.sh ; /setup/preprocess/standalone-preprocess.py
LLMDBENCH_VLLM_STANDALONE_ROUTE
LLMDBENCH_VLLM_STANDALONE_HTTPROUTE
LLMDBENCH_VLLM_STANDALONE_ARGS
LLMDBENCH_VLLM_STANDALONE_EPHEMERAL_STORAGE
  • "llm-d"-specific VLLM paramaters
Variable Meaning Note
LLMDBENCH_VLLM_INFRA_CHART_NAME
LLMDBENCH_VLLM_INFRA_CHART_VERSION
LLMDBENCH_VLLM_INFRA_GATEWAY_CPU_REQUEST Gateway CPU request Default=4
LLMDBENCH_VLLM_INFRA_GATEWAY_CPU_LIMIT Gateway CPU limit Default=16
LLMDBENCH_VLLM_INFRA_GATEWAY_MEMORY_REQUEST Gateway memory request Default=4Gi
LLMDBENCH_VLLM_INFRA_GATEWAY_MEMORY_LIMIT Gateway memory limit Default=16Gi
LLMDBENCH_VLLM_GAIE_CHART_NAME
LLMDBENCH_VLLM_GAIE_CHART_VERSION
LLMDBENCH_VLLM_MODELSERVICE_RELEASE
LLMDBENCH_VLLM_MODELSERVICE_VALUES_FILE
LLMDBENCH_VLLM_MODELSERVICE_ADDITIONAL_SETS
LLMDBENCH_VLLM_MODELSERVICE_CHART_VERSION
LLMDBENCH_VLLM_MODELSERVICE_CHART_NAME
LLMDBENCH_VLLM_MODELSERVICE_HELM_REPOSITORY
LLMDBENCH_VLLM_MODELSERVICE_HELM_REPOSITORY_URL
LLMDBENCH_VLLM_MODELSERVICE_URI_PROTOCOL
LLMDBENCH_VLLM_MODELSERVICE_DECODE_INFERENCE_PORT
LLMDBENCH_VLLM_MODELSERVICE_GATEWAY_CLASS_NAME
LLMDBENCH_VLLM_MODELSERVICE_ROUTE
LLMDBENCH_VLLM_MODELSERVICE_EPP
LLMDBENCH_VLLM_MODELSERVICE_INFERENCE_MODEL
LLMDBENCH_VLLM_MODELSERVICE_INFERENCE_POOL
LLMDBENCH_VLLM_MODELSERVICE_GAIE_PLUGINS_CONFIGFILE
LLMDBENCH_VLLM_MODELSERVICE_GAIE_MONITORING_PROMETHEUS_ENABLED Enable Prometheus ServiceMonitor for GAIE EPP component metrics true (default) or false false