From 30297b8f30b5adaf4cd9b8d7a1532d1369c00070 Mon Sep 17 00:00:00 2001 From: Yufeng Zhu Date: Sun, 19 Apr 2026 13:43:49 +0800 Subject: [PATCH] Relax pixel tolerance to 15 to account for Triton kernel numerical drift Signed-off-by: Yufeng Zhu Signed-off-by: princepride --- .../omni_connectors/test_bagel_shared_memory_connector.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/distributed/omni_connectors/test_bagel_shared_memory_connector.py b/tests/distributed/omni_connectors/test_bagel_shared_memory_connector.py index 57c6a79fb..611d3e9c0 100644 --- a/tests/distributed/omni_connectors/test_bagel_shared_memory_connector.py +++ b/tests/distributed/omni_connectors/test_bagel_shared_memory_connector.py @@ -4,7 +4,7 @@ """ End-to-end tests for Bagel with shared memory connector: img2img and text2img. -- img2img: validates output vs reference pixels within a ±10 tolerance. +- img2img: validates output vs reference pixels within a ±15 tolerance. - text2img: validates output vs reference pixels within a ±5 tolerance (equivalent to `examples/offline_inference/bagel/end2end.py` with text2img modality and 15 steps). @@ -91,9 +91,11 @@ if current_omni_platform.is_rocm(): {"position": (256, 256), "rgb": (167, 156, 150)}, ] -PIXEL_TOLERANCE = 10 TEXT2IMG_PIXEL_TOLERANCE = 5 +# Relax pixel tolerance to 15 to account for Triton kernel numerical drift +PIXEL_TOLERANCE = 15 + DEFAULT_PROMPT = "<|fim_middle|><|im_start|>Change the grass color to red<|im_end|>" TEXT2IMG_DEFAULT_PROMPT = "<|im_start|>A cute cat<|im_end|>"