Skip to content

Conversation

@rkarhila-amd
Copy link
Collaborator

A small but significantly performance increasing change. MI300 and MI325 to follow after testing. Slurm changes untested.

@rkarhila-amd rkarhila-amd requested a review from a team as a code owner November 19, 2025 14:27
@functionstackx
Copy link
Contributor

@rkarhila-amd thanks for the PR. did u happen to change the wrong thing in amd-master.yaml, i see you editted only fp8 bash scripts and ur title says 'dsr1 fp8 image' but then u only editted dsr1 fp4`

image

austenstone and others added 9 commits December 3, 2025 16:33
* Adjust preview for dark mode and light mode

picture element for better display based on color scheme.

* Rounded

* Update image alt text in README.md
Co-authored-by: Jatin Gangani <jgangani@dc2-container-xterm-014.prd.it.nvidia.com>
* initial poc

* remove -d flag when launching docker container

* syntax error

* compatibility fixes

* add correct endpoint prefix

* remove reference env var

* run vllm serve in background

* unescape sequences

* stop vllm to stdout after it stops

* stop vllm to stdout after it stops pt 2

* get rid of docker stop as no longer in detatched

* clone bench serving to tmp dir

* clone bench serving to tmp dir pt 2

* add explanatory comment

* cleaning up

* cleaning up

* adding mi355x refactor

* adding h200 initial refactor

* different way to see server logs

* cleanup

* now fail if server fails

* starting on b200

* doign b200

* reverting erroneous change

* fixing b200

* fixing b200 pt 2

* updating mi300

* updating mi300 pt 2

* updating mi300 pt 3 -- remove detached mode

* cleaning up mi355x

* fixing mi300x and updating 325x

* reverting max conc to 512 on gptoss fp4 b200 docker

* fixing mi300x and updating 325x

* cleanng up

* add wait for h200 slurm dsr1

* max num seqs back to 512 for gptoss fpr b200 docker

* fix port issue for dsr1 mi300x docker

* fix mi355x docker NUM_PROMPTS

* adding prop of failure for server logs

* add utils function for benchmark

* add utils function for benchmark

* function-ize the waiting for server to start

* dont show arg parsing set -x

* dont show arg parsing set +x oops

* dont show arg parsing set +x oops

* capture server pid

* nebdius dont scancel

* changes to comments in benchmark lib . sh

* Update benchmarks/dsr1_fp4_mi355x_docker.sh

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

* Update .github/workflows/benchmark-tmpl.yml

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

* adding back whitespace

* adding back whitespace

* adding back whitespace

* remove tg launch script

* Update benchmarks/gptoss_fp4_h100_docker.sh

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

* Update benchmarks/dsr1_fp8_mi325x_docker.sh

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

* Update benchmarks/dsr1_fp8_mi355x_docker.sh

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

* Update benchmarks/gptoss_fp4_b200_trt_slurm.sh

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

* Audit and correct required environment variables documentation in all benchmark scripts (InferenceMAX#252)

* Initial plan

* Update required env vars documentation in all benchmark scripts

Co-authored-by: cquil11 <60715037+cquil11@users.noreply.github.com>

* Fix required env vars - remove NF, PREFILL_SIZE, and correct PORT/PORT_OFFSET

Co-authored-by: cquil11 <60715037+cquil11@users.noreply.github.com>

* Remove internally-calculated vars from required env vars (EXTRA_CONFIG_FILE, MAX_NUM_TOKENS, MOE_BACKEND)

Co-authored-by: cquil11 <60715037+cquil11@users.noreply.github.com>

---------

Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: cquil11 <60715037+cquil11@users.noreply.github.com>

* removing oci node rebase with main

---------

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Co-authored-by: Copilot <198982749+Copilot@users.noreply.github.com>
…InferenceMAX#253)

Bumps the github-actions group with 1 update: [actions/checkout](https://github.com/actions/checkout).


Updates `actions/checkout` from 5.0.0 to 6.0.0
- [Release notes](https://github.com/actions/checkout/releases)
- [Changelog](https://github.com/actions/checkout/blob/main/CHANGELOG.md)
- [Commits](actions/checkout@08c6903...1af3b93)

---
updated-dependencies:
- dependency-name: actions/checkout
  dependency-version: 6.0.0
  dependency-type: direct:production
  update-type: version-update:semver-major
  dependency-group: github-actions
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Bumps the github-actions group with 1 update: [actions/setup-python](https://github.com/actions/setup-python).


Updates `actions/setup-python` from 6.0.0 to 6.1.0
- [Release notes](https://github.com/actions/setup-python/releases)
- [Commits](actions/setup-python@e797f83...83679a8)

---
updated-dependencies:
- dependency-name: actions/setup-python
  dependency-version: 6.1.0
  dependency-type: direct:production
  update-type: version-update:semver-minor
  dependency-group: github-actions
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
* Bumps DSR1 SGLang code

* update how we get the resulting log files

---------

Co-authored-by: Elnifio <elnifio0519@gmail.com>
Co-authored-by: Cameron Quilici <cjquilici@gmail.com>
* Update GPTOSS B200 AGG

* set dp attention env vars

* Add DP attn comment

---------

Co-authored-by: Jatin Gangani <jgangani@dc2-container-xterm-014.prd.it.nvidia.com>
@rkarhila-amd
Copy link
Collaborator Author

did u happen to change the wrong thing in

Yeah, sorry, my bad!

We're now updating this PR to include some changes to runners to enable FP4 on the same lmsys container.

export ROCM_QUICK_REDUCE_QUANTIZATION=INT4


SERVER_LOG=$(mktemp /tmp/server-XXXXXX.log)
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

dupe

export RCCL_MSCCL_ENABLE=0
export ROCM_QUICK_REDUCE_QUANTIZATION=INT4


Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nit whitespace

@cquil11 cquil11 temporarily deployed to fork-pr-validation December 3, 2025 17:27 — with GitHub Actions Inactive
@cquil11
Copy link
Collaborator

cquil11 commented Dec 3, 2025

old perf

Model Hardware Framework Precision ISL OSL TP EP DP Attention Conc TTFT (ms) TPOT (ms) Interactivity (tok/s/user) E2EL (s) TPUT per GPU Output TPUT per GPU Input TPUT per GPU
deepseek-ai/DeepSeek-R1-0528 MI355X SGLANG FP8 1024 1024 8 1 false 4 99.1861 18.5055 54.0381 17.1707 53.4397 26.7035 26.7362

https://github.com/InferenceMAX/InferenceMAX/actions/runs/19776095504#:~:text=206.2476-,deepseek%2Dai/DeepSeek%2DR1%2D0528,26.7362,-deepseek%2Dai/DeepSeek

new perf

Model Hardware Framework Precision ISL OSL TP EP DP Attention Conc TTFT (ms) TPOT (ms) Interactivity (tok/s/user) E2EL (s) TPUT per GPU Output TPUT per GPU Input TPUT per GPU
deepseek-ai/DeepSeek-R1-0528 MI355X SGLANG FP8 1024 1024 8 1 false 4 91.5310 18.4025 54.3404 17.0457 53.7069 26.8370 26.8698

https://github.com/InferenceMAX/InferenceMAX/actions/runs/19902871108?pr=247#:~:text=deepseek%2Dai/DeepSeek,26.8698

looks good to me, if all good with you? @rkarhila-amd
please just address comments and we should be good to go

@cquil11
Copy link
Collaborator

cquil11 commented Dec 3, 2025

pls also correct PR description

@rkarhila-amd rkarhila-amd changed the title Change dsr1 fp8 image to lmsysorg/sglang 0.5.5.post2 for AMD MI355 Change dsr1 fp8 image to lmsysorg/sglang 0.5.5.post3 and fp4 image to 0.5.5.post2 for AMD MI355 Dec 4, 2025
@rkarhila-amd
Copy link
Collaborator Author

Thanks @cquil11 , good to go from my side!

@cquil11 cquil11 temporarily deployed to fork-pr-validation December 4, 2025 23:39 — with GitHub Actions Inactive
@cquil11 cquil11 enabled auto-merge (squash) December 5, 2025 00:02
Copy link
Collaborator

@cquil11 cquil11 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm, thank you!

@cquil11 cquil11 merged commit 0fe9dcc into InferenceMAX:main Dec 5, 2025
3 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

8 participants