20231088/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[Bugfix][Doc] Report neuron error in output (#9159)

Browse Source
This commit is contained in:
Joe Rowell 2024-10-09 06:43:34 +01:00 committed by GitHub
parent 480b7f40cf
commit 7627172bf4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@ -332,7 +332,7 @@ def get_neuronxcc_version():
# Return the version string # Return the version string
return match.group(1) return match.group(1)
else: else:
raise RuntimeError("Could not find HIP version in the output") raise RuntimeError("Could not find Neuron version in the output")
def get_nvcc_cuda_version() -> Version: def get_nvcc_cuda_version() -> Version: