Repository navigation
[Klaud Cold] feat(ubenchX): add MI355X, MI325X, MI300X mem_bw results / [Klaud Cold] feat(ubenchX):新增 MI355X、MI325X、MI300X mem_bw 测试结果 - #3880
Conversation
中文:新增 MI355X、MI325X、MI300X 的 mem_bw 测试结果 Co-Authored-By: Claude Opus 5.5 (1M context) <noreply@anthropic.com>
There was a problem hiding this comment.
Looks good — straightforward data-only addition, no code changes.
Reviewed: the three new CSV files (mi300x/mi325x/mi355x) each contain the expected header plus all 32 power-of-two sizes from 8 B to 16 GiB. Spot-checked the bandwidth_gbps column against bytes*2/(time_ms*1e-3)/1e9 (the formula in ubenchX/mem_bw/bench.py) and it matches exactly, and the peak/16 GiB figures match those quoted in the PR description.
Extended reasoning...
Diff only adds three new CSV result files (99 lines total, no code/scripts/configs touched) under ubenchX/mem_bw/results/ for MI300X/MI325X/MI355X, mirroring the sibling NVIDIA-result commit already in the base. No security-sensitive surface is touched. Verified the bandwidth values in all three files are internally consistent with the bench.py formula (size*2/time) and match the peak/16 GiB numbers claimed in the PR description. Only generic wildcard CODEOWNERS (@ SemiAnalysisAI/core) covers this path, no specific owner objection, and no prior review history exists on this PR.
Add
ubenchX/mem_bwresults for AMD MI355X, MI325X, and MI300X. Each CSV has all 32 sizes (8 B to 16 GiB) from one GPU, produced byubenchX/mem_bw/bench.pyand copied from its printed output.mi355x_2026-10-09.csvsglang_local-rocm724-mi35x.sqsh(cached local SGLang ROCm 7.2.4 build)mi325x_2026-10-09.csvrocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0mi300x_2026-10-09.csvrocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0Clusters: MI355X on the TensorWave MI355X cluster (
mia1-p01-g50); MI325X and MI300X on the AMDccs1salogin01cluster (MI325X-UBUNTU,MI300X-UBUNTU).Notes: on AMD, bandwidth peaks around 1–4 GiB and dips slightly at the largest sizes, while the NVIDIA runs keep rising to 16 GiB.
bench.pyran unchanged; inside the containersTRITON_CACHE_DIRandHOMEpointed at the writable mount because the default home was read-only. All runs are single samples on a single GPU, and container/torch versions differ across GPUs.中文
新增 AMD MI355X、MI325X 和 MI300X 的
ubenchX/mem_bw结果。每个 CSV 均来自单块 GPU,包含 8 B 至 16 GiB 全部 32 个大小,由ubenchX/mem_bw/bench.py生成,并按其打印输出整理。mi355x_2026-10-09.csvsglang_local-rocm724-mi35x.sqsh(已缓存的本地 SGLang ROCm 7.2.4 构建)mi325x_2026-10-09.csvrocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0mi300x_2026-10-09.csvrocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0集群:MI355X 运行于 TensorWave MI355X 集群(
mia1-p01-g50);MI325X 和 MI300X 运行于 AMDccs1salogin01集群(MI325X-UBUNTU、MI300X-UBUNTU)。说明:在 AMD 上,带宽在 1–4 GiB 附近达到峰值,最大几个大小处略有回落;而 NVIDIA 的结果会一直上升到 16 GiB。
bench.py未做修改;由于容器内默认 home 目录只读,TRITON_CACHE_DIR和HOME被指向可写的挂载目录。所有结果均为单块 GPU 上的单次测量,各 GPU 的容器和 torch 版本也不相同。AI model disclosure
claude-opus-5-5[1m](Claude Opus 5.5, 1M context) — ran the benchmarks on the clusters, compiled the results, and wrote this PR.🤖 Generated with Claude Code