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

Don't add CUDA version to version in case not for CUDA

Browse source 
Signed-off-by: fxzjshm <fxzjshm@163.com>
This commit is contained in:
fxzjshm 2025-02-13 00:59:28 +08:00
parent 21fca5a326
commit 4cda45433f
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
setup.py
View file

@ -129,7 +129,7 @@ class VersionInfo:
def get_package_version(self, full_version=False):
flash_version = self.get_flash_version()
package_version = f"{str(flash_version)}+cu{self.get_cuda_bare_metal_version(CUDA_HOME)}torch{self.get_torch_version()}{self.get_cpu_instruct()}"
package_version = f"{str(flash_version)}+torch{self.get_torch_version()}{self.get_cpu_instruct()}"
if full_version:
return package_version
if not VersionInfo.FORCE_BUILD:
@ -306,7 +306,7 @@ setup(
'cxx': ['-O3'],
'nvcc': [
'-O3',
'--use_fast_math',
# '--use_fast_math',
'-Xcompiler', '-fPIC',
]
}