From 4cda45433f5d464fcddba0eebb69175acf71eee1 Mon Sep 17 00:00:00 2001 From: fxzjshm Date: Thu, 13 Feb 2025 00:59:28 +0800 Subject: [PATCH] Don't add CUDA version to version in case not for CUDA Signed-off-by: fxzjshm --- setup.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index d24db14b..390fec6a 100644 --- a/setup.py +++ b/setup.py @@ -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', ] }