From a578bf814cd4b8decd1ee0f1190217bf1b76ef72 Mon Sep 17 00:00:00 2001 From: billishyahao Date: Fri, 1 May 2026 14:36:20 +0800 Subject: [PATCH] [AMD] fix moriep unittest failure (#24205) --- test/registered/amd/test_moriep_small.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/test/registered/amd/test_moriep_small.py b/test/registered/amd/test_moriep_small.py index 42547cf9e..0d073b5f3 100644 --- a/test/registered/amd/test_moriep_small.py +++ b/test/registered/amd/test_moriep_small.py @@ -400,6 +400,7 @@ class TestMTPwithTBONormal(CustomTestCase): env["SGLANG_MORI_DISPATCH_DTYPE"] = "bf16" env["SGLANG_MORI_NUM_MAX_DISPATCH_TOKENS_PER_RANK"] = "4096" env["MORI_SHMEM_MODE"] = "ISOLATION" # avoid out of symmetric heap memory + env["SGLANG_ENABLE_SPEC_V2"] = "false" cls.process = popen_launch_server( cls.model, @@ -458,6 +459,7 @@ class TestMTPwithTBOLowLatency(CustomTestCase): env["SGLANG_USE_AITER"] = "1" env["SGLANG_MORI_DISPATCH_DTYPE"] = "bf16" env["SGLANG_MORI_NUM_MAX_DISPATCH_TOKENS_PER_RANK"] = "4096" + env["SGLANG_ENABLE_SPEC_V2"] = "false" env["MORI_SHMEM_MODE"] = "ISOLATION" # avoid out of symmetric heap memory # FIXME(billishyahao): enable p2p due to no rdma devices on CI machine # env["MORI_DISABLE_P2P"] = "1"