[ROCm][CI] Keep rocprofiler-sdk out of DeepEP HT MoE test workers (#51173)

Signed-off-by: Stefan Koncarevic <Stefan.Koncarevic@amd.com>
Co-authored-by: Andreas Karatzas <akaratza@amd.com>
This commit is contained in:
stefankoncarevic
2026-08-05 15:28:41 -05:00
committed by GitHub
co-authored by Andreas Karatzas
parent c2d8009044
commit 65addac701
+7
View File
@@ -488,10 +488,17 @@ def test_deep_ep_moe(
world_dp_size: tuple[int, int],
per_act_token_quant: bool,
workspace_init,
monkeypatch,
):
low_latency_mode = False
use_fp8_dispatch = False
if current_platform.is_rocm():
# The cooperative kernel launches on this path segfault inside the ROCm
# HSA runtime at process exit while rocprofiler-sdk is attached, and
# torch attaches it implicitly. Workers are spawned, so they inherit it.
monkeypatch.setenv("ROCPROFILER_REGISTER_ENABLED", "0")
set_random_seed(7)
world_size, dp_size = world_dp_size
config = TestConfig(dtype=dtype, topk=topk, m=m, k=k, n=n, num_experts=num_experts)