"vllm/vscode:/vscode.git/clone" did not exist on "54de71d0dfbb6340fdbc620f4ebeb4236d165a37"
Commit 4b4eeb26 authored by zhuwenwen's avatar zhuwenwen
Browse files

Merge remote-tracking branch 'mirror/main'

parents 2216a4e5 4fdc581f
...@@ -5,10 +5,12 @@ from vllm.platforms import current_platform ...@@ -5,10 +5,12 @@ from vllm.platforms import current_platform
logger = init_logger(__name__) logger = init_logger(__name__)
# neuron has too old torch HAS_TRITON = (
HAS_TRITON = find_spec( find_spec("triton") is not None
"triton") is not None and not current_platform.is_neuron() and not current_platform.is_xpu() # Not compatible
and not current_platform.is_neuron() # neuron has too old torch
)
if not HAS_TRITON: if not HAS_TRITON:
logger.info("Triton not installed; certain GPU-related functions" logger.info("Triton not installed or not compatible; certain GPU-related"
" will not be available.") " functions will not be available.")
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
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