[CI][ROCm] Route LTX2 Ulysses parity to two-GPU lane (#7234)

Signed-off-by: andyluo7 <andy.luo@amd.com>
This commit is contained in:
andyluo7 authored and GitHub committed 2026-09-08 22:59:37 +08:00
1 parent 7422a85cc7
commit de1b0708bd
2 files changed
+17 -2

No files matched your search

@@ -156,10 +156,14 @@ def _run_transformer_parity(rank: int, world_size: int, master_port: int) -> Non
@pytest.mark.full_model
@hardware_test(res={"cuda": "L4"}, num_cards=2)
@hardware_test(res={"cuda": "L4", "rocm": "MI325"}, num_cards=2)
def test_ltx_ulysses_transformer_matches_sp1(unused_tcp_port) -> None:
"""Exercise real plan hooks, RoPE/timestep sharding, and attention routing."""
world_size = 2
device_count = current_omni_platform.device_count()
assert device_count >= world_size, (
f"LTX Ulysses parity requires {world_size} accelerator devices; only {device_count} are visible"
)
torch.multiprocessing.spawn(
_run_transformer_parity,
args=(world_size, unused_tcp_port),