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

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

View File

@ -332,7 +332,7 @@ def get_neuronxcc_version():
# Return the version string
return match.group(1)
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: