Commit 4cda4543 authored by fxzjshm's avatar fxzjshm
Browse files

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


Signed-off-by: fxzjshm's avatarfxzjshm <fxzjshm@163.com>
parent 21fca5a3
......@@ -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',
]
}
......
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