[Bugfix] Fix ROCm crash in qwen3_next multi-stream events (#36795) (#37427)

Signed-off-by: JartX <sagformas@epdcenter.es>
This commit is contained in:
JartX
2026-03-18 21:06:31 +01:00
committed by GitHub
parent 5ce2d10e4a
commit a913b612d8

View File

@@ -427,7 +427,7 @@ class Qwen3NextGatedDeltaNet(nn.Module, MambaBase):
self.aux_stream = aux_stream()
self.events = (
[torch.cuda.Event(), torch.cuda.Event()]
if current_platform.is_cuda()
if current_platform.is_cuda_alike()
else [None, None]
)