From 0c2464db200f4b0046222fe7be2f571b8d250faf Mon Sep 17 00:00:00 2001 From: Taichang Zhou Date: Sat, 7 Mar 2026 13:26:32 +0800 Subject: [PATCH] revert unnecessary changes Signed-off-by: Taichang Zhou --- tests/diffusion/attention/test_attention_sp.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/diffusion/attention/test_attention_sp.py b/tests/diffusion/attention/test_attention_sp.py index e261d7dcb..234ac1563 100644 --- a/tests/diffusion/attention/test_attention_sp.py +++ b/tests/diffusion/attention/test_attention_sp.py @@ -329,9 +329,7 @@ def ulysses_attention_on_test_model( """Run Ulysses attention test on a test model and save results for comparison.""" # Use fixed seed for reproducibility across baseline and SP runs RANDOM_SEED = 42 - torch.manual_seed(RANDOM_SEED) - if torch.cuda.is_available(): - torch.cuda.manual_seed_all(RANDOM_SEED) + current_omni_platform.seed_everything(RANDOM_SEED) mode_str = "Baseline (no SP)" if is_baseline else f"SP (ulysses={ulysses_degree}, ring={ring_degree})" print(f"\n[{mode_str}] Rank {local_rank}/{world_size} - Random seed set to {RANDOM_SEED}")