[CI][Bugfix] Relax stable-audio layerwise offload determinism tolerance to 1e-2 (#3371)

Signed-off-by: Yueqian Lin <linyueqian@outlook.com>
This commit is contained in:
Yueqian Lin authored and GitHub committed 2026-05-06 11:32:39 +08:00
1 parent 005621ba7f
commit 354511b805
1 file changed
+4 -1
@@ -98,7 +98,10 @@ def test_layerwise_offload_diffusion_model(model_name: str):
if model_name == "stabilityai/stable-audio-open-1.0":
audio_offload = output_offload[0].request_output.multimodal_output.get("audio")
audio_no_offload = output_no_offload[0].request_output.multimodal_output.get("audio")
check_audio_determinism(audio_offload, audio_no_offload, atol=1e-3)
# Match the sibling cpu-offload test's tolerance: layerwise offload moves
# blocks across the PCIe bus on a side stream, which can perturb cuBLAS
# algorithm selection and produce ~ULP-level drift larger than 1e-3.
check_audio_determinism(audio_offload, audio_no_offload, atol=1e-2)
is_rocm = torch.version.hip is not None
platform = "rocm" if is_rocm else "cuda"