update rocm Stream

236f11df · zhuwenwen · 77a29416 · 236f11df
Commit 236f11df authored Aug 07, 2025 by zhuwenwen
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

vllm/utils/__init__.py vllm/utils/__init__.py +3 -3

No files found.
--- a/vllm/utils/__init__.py
+++ b/vllm/utils/__init__.py
@@ -1428,9 +1428,9 @@ def current_stream() -> torch.cuda.Stream:
        # per process
        # fix computational precision issue in eager mode 
-        # _current_stream_tls.value = torch.cuda.Stream(
+        _current_stream_tls.value = torch.cuda.Stream(
-        # ) if current_platform.is_rocm() else torch.cuda.current_stream()
+        ) if current_platform.is_rocm() else torch.cuda.current_stream()
-        _current_stream_tls.value = torch.cuda.current_stream()
+        # _current_stream_tls.value = torch.cuda.current_stream()
    return _current_stream_tls.value