Skip to content

GitLab

Commit bdda5719 authored by zhuwenwen's avatar zhuwenwen
Browse files

Revert "update rocm Stream" 

This reverts commit 236f11df.
parent 71b02b7a
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
vllm/utils/__init__.py
View file @ bdda5719
...@@ -1428,9 +1428,9 @@ def current_stream() -> torch.cuda.Stream: ...@@ -1428,9 +1428,9 @@ def current_stream() -> torch.cuda.Stream:
# per process # per process
# fix computational precision issue in eager mode # 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 return _current_stream_tls.value
......
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