vrr/kvcache-ai-ktransformers
2
0
Fork 0
mirror of https://github.com/kvcache-ai/ktransformers.git synced 2025-09-05 12:09:48 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

fix: refine backend error message to include ROCM_HOME

Browse source 
Signed-off-by: fishingfly <zhoyuzf@163.com>
This commit is contained in:
fishingfly 2025-04-01 10:50:38 +08:00
parent f142f4dff3
commit 7549ff335a
1 changed files with 9 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
setup.py
View file

@ -316,7 +316,7 @@ class CMakeBuild(BuildExtension):
elif ROCM_HOME is not None: elif ROCM_HOME is not None:
cmake_args += ["-DKTRANSFORMERS_USE_ROCM=ON"] cmake_args += ["-DKTRANSFORMERS_USE_ROCM=ON"]
else: else:
raise ValueError("Unsupported backend: CUDA_HOME and MUSA_HOME are not set.") raise ValueError("Unsupported backend: CUDA_HOME, MUSA_HOME, and ROCM_HOME are not set.")
# log cmake_args # log cmake_args
print("CMake args:", cmake_args) print("CMake args:", cmake_args)