Unverified Commit a913b612 authored by JartX's avatar JartX Committed by GitHub
Browse files

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


Signed-off-by: default avatarJartX <sagformas@epdcenter.es>
parent 5ce2d10e
......@@ -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]
)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment