Unverified Commit 45a4b7cf authored by Yan Ru Pei's avatar Yan Ru Pei Committed by GitHub
Browse files

docs: mention need for H100 for gpt-oss-120b in router benchmarking docs (#3242)


Signed-off-by: default avatarYan Ru Pei <yanrpei@gmail.com>
Co-authored-by: default avatarAlec <35311602+alec-flowers@users.noreply.github.com>
parent a0f845e3
<!-- # SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. <!-- # SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0 # SPDX-License-Identifier: Apache-2.0 -->
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License. -->
# Router Benchmarking Guide # Router Benchmarking Guide
...@@ -20,6 +8,7 @@ This directory contains scripts for benchmarking the Dynamo router with prefix c ...@@ -20,6 +8,7 @@ This directory contains scripts for benchmarking the Dynamo router with prefix c
## Prerequisites ## Prerequisites
- NVIDIA GPUs (8 GPUs for default configuration) - NVIDIA GPUs (8 GPUs for default configuration)
- (optional) H100 GPUs or later for gpt-oss-120b examples
- CUDA environment properly configured - CUDA environment properly configured
- etcd and NATS running (required for Dynamo coordination) - etcd and NATS running (required for Dynamo coordination)
- Required Python packages: - Required Python packages:
...@@ -60,7 +49,7 @@ Make sure you have 8 GPUs for these examples, unless you are using mockers (see ...@@ -60,7 +49,7 @@ Make sure you have 8 GPUs for these examples, unless you are using mockers (see
--model-path deepseek-ai/DeepSeek-R1-Distill-Llama-8B --model-path deepseek-ai/DeepSeek-R1-Distill-Llama-8B
# Example: 4 vLLM workers with larger model using tensor parallelism (2 GPUs per worker) # Example: 4 vLLM workers with larger model using tensor parallelism (2 GPUs per worker)
# NOTE: this would likely require each GPU having 80GB of VRAM # NOTE: this requires having Hopper or later GPU SKUs to support MXFP4 precision.
./run_engines.sh \ ./run_engines.sh \
--num-workers 4 \ --num-workers 4 \
--model-path openai/gpt-oss-120b \ --model-path openai/gpt-oss-120b \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment