Skip to content

GitLab

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
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
setup.py
View file @ 4cda4543
...@@ -129,7 +129,7 @@ class VersionInfo: ...@@ -129,7 +129,7 @@ class VersionInfo:
def get_package_version(self, full_version=False): def get_package_version(self, full_version=False):
flash_version = self.get_flash_version() 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: if full_version:
return package_version return package_version
if not VersionInfo.FORCE_BUILD: if not VersionInfo.FORCE_BUILD:
...@@ -306,7 +306,7 @@ setup( ...@@ -306,7 +306,7 @@ setup(
'cxx': ['-O3'], 'cxx': ['-O3'],
'nvcc': [ 'nvcc': [
'-O3', '-O3',
'--use_fast_math', # '--use_fast_math',
'-Xcompiler', '-fPIC', '-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