diff --git a/setup.py b/setup.py index 67880bee465ca32e7d83bf1f3d6e1311710b596e..6dea124dd06bbcde9d15a17b282b62a00bb6d2f2 100644 --- a/setup.py +++ b/setup.py @@ -44,11 +44,11 @@ def get_sha(pytorch_root: Union[str, Path]) -> str: return "Unknown" BASE_DIR = os.path.dirname(os.path.realpath(__file__)) -VERSION = torch.__version__ +VERSION = "1.0-" + torch.__version__[0:6] torch_npu_root = Path(__file__).parent sha = get_sha(torch_npu_root) -if os.getenv("BUILD_WITHOUT_SHA") is None: - VERSION += "_" + sha[:7] +if not os.getenv("BUILD_WITHOUT_SHA"): + VERSION += "+git" + sha[:7] setup( name="ads_accelerator",