[AMD] Skip test_update_weights_from_disk on ROCm pending reload fix (#31924) (#31925)

This commit is contained in:
kangwangamd
2026-07-30 02:32:09 -07:00
committed by GitHub
parent fc007e1f00
commit 4b52758c76
@@ -997,6 +997,20 @@ STANDALONE_FILES = {
],
}
# test_update_weights_from_disk fails deterministically on ROCm: the diffusion
# weight-update-from-disk reload path does not reconcile the diffusers
# checkpoint layout with the sglang transformer parameters (shape mismatch ->
# HTTP 400). Tracked in #31924. NVIDIA does not run standalone files, so this
# test is AMD-only; skip it on ROCm until the reload path is fixed. Remove
# this block (keeping the STANDALONE_FILES / est-time entry above) to
# re-enable once #31924 lands.
if current_platform.is_hip():
STANDALONE_FILES["1-gpu"] = [
f
for f in STANDALONE_FILES["1-gpu"]
if f != "../single_test_file/test_update_weights_from_disk.py"
]
# New standalone files may omit an estimate once to learn the real CI runtime.
# CI will use a fallback estimate for sharding, run the test, then print a
# measured value that must be copied into STANDALONE_FILE_EST_TIMES.